From 0fd13fa41046d30d75c6f8a6dc9b542f9cee582f Mon Sep 17 00:00:00 2001 From: LDOUBLEV Date: Wed, 26 May 2021 10:54:54 +0000 Subject: [PATCH] process else --- tools/infer/utility.py | 5 ++++- 1 file changed, 4 insertions(+), 1 deletion(-) diff --git a/tools/infer/utility.py b/tools/infer/utility.py index ff0b60b9..de3bb3db 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) -- GitLab