fix dy shapes of trt for rec models

PaddlePaddle · Feb 10, 2025 · a5294be · a5294be
1 parent 2376b03
commit a5294be
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/ppocr/utils/export_model.py b/ppocr/utils/export_model.py
@@ -49,7 +49,7 @@ def dump_infer_config(config, path, logger):
         arch_config = config["Architecture"]
         if arch_config["algorithm"] in ["SVTR_LCNet", "SVTR_HGNet"]:
             common_dynamic_shapes = {
-                "x": [[1, 3, 24, 160], [1, 3, 48, 320], [8, 3, 96, 640]]
+                "x": [[1, 3, 48, 160], [1, 3, 48, 320], [8, 3, 48, 640]]
             }
         elif arch_config["model_type"] == "det":
             common_dynamic_shapes = {