Merge pull request #4845 from andyjpaddle/fix_trt_dynamic_shape

fix rec trt dynamic shape error

tools/infer/utility.py

@@ -271,8 +271,13 @@ def create_predictor(args, mode, logger):
                 min_input_shape = {"x": [1, 3, 10, 10]}
                 max_input_shape = {"x": [1, 3, 512, 512]}
                 opt_input_shape = {"x": [1, 3, 256, 256]}
-            config.set_trt_dynamic_shape_info(min_input_shape, max_input_shape,
-                                              opt_input_shape)
+            if mode == "rec":
+                if args.rec_algorithm == "CRNN":
+                    config.set_trt_dynamic_shape_info(
+                        min_input_shape, max_input_shape, opt_input_shape)
+            else:
+                config.set_trt_dynamic_shape_info(
+                    min_input_shape, max_input_shape, opt_input_shape)
 
         else:
             config.disable_gpu()