diff --git a/tools/export_model.py b/tools/export_model.py
index c7c1158ed1e5f02d9362de7307bf1e1874fd6d63..cf9df78dd03eb52b0e0790e1bb2c32e09a1faeb6 100755
--- a/tools/export_model.py
+++ b/tools/export_model.py
@@ -130,7 +130,7 @@ def export_single_model(model,
     else:
         infer_shape = [3, -1, -1]
         if arch_config["model_type"] == "rec":
-            infer_shape = [3, 48, -1]  # for rec model, H must be 32
+            infer_shape = [3, 32, -1]  # for rec model, H must be 32
             if "Transform" in arch_config and arch_config[
                     "Transform"] is not None and arch_config["Transform"][
                         "name"] == "TPS":