From c39274040f98c1b5cf19f75da4e5b16c93203b46 Mon Sep 17 00:00:00 2001 From: LDOUBLEV Date: Tue, 30 Nov 2021 11:30:21 +0000 Subject: [PATCH] larger dynamic shape --- tools/infer/utility.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/tools/infer/utility.py b/tools/infer/utility.py index c2881665..f437056e 100644 --- a/tools/infer/utility.py +++ b/tools/infer/utility.py @@ -261,7 +261,7 @@ def create_predictor(args, mode, logger): opt_input_shape.update(opt_pact_shape) elif mode == "rec": min_input_shape = {"x": [1, 3, 32, 10]} - max_input_shape = {"x": [args.rec_batch_num, 3, 32, 1024]} + max_input_shape = {"x": [args.rec_batch_num, 3, 32, 1536]} opt_input_shape = {"x": [args.rec_batch_num, 3, 32, 320]} elif mode == "cls": min_input_shape = {"x": [1, 3, 48, 10]} -- GitLab