diff --git a/tools/infer/utility.py b/tools/infer/utility.py
index ff0b60b9ccfe3361a5dd53e3c0f4420730cf814a..de3bb3db9742859ed2f098d90c87c1610e0d7cfe 100755
--- a/tools/infer/utility.py
+++ b/tools/infer/utility.py
@@ -212,7 +212,10 @@ def create_predictor(args, mode, logger):
             min_input_shape = {"x": [args.rec_batch_num, 3, 48, 10]}
             max_input_shape = {"x": [args.rec_batch_num, 3, 48, 2000]}
             opt_input_shape = {"x": [args.rec_batch_num, 3, 48, 320]}
-
+        else:
+            min_input_shape = {"x": [1, 3, 10, 10]}
+            max_input_shape = {"x": [1, 3, 1000, 1000]}
+            opt_input_shape = {"x": [1, 3, 500, 500]}
         config.set_trt_dynamic_shape_info(min_input_shape, max_input_shape,
                                           opt_input_shape)