From b70255afabd2cc12a158a12a34e0e5088d53ed00 Mon Sep 17 00:00:00 2001 From: Armin Date: Thu, 16 Jun 2022 15:11:59 +0200 Subject: [PATCH] Fix TensorRT failed due to some TRT inputs dynamic shape info not set --- tools/infer/utility.py | 1 + 1 file changed, 1 insertion(+) diff --git a/tools/infer/utility.py b/tools/infer/utility.py index bc095baa..4f4caaba 100644 --- a/tools/infer/utility.py +++ b/tools/infer/utility.py @@ -276,6 +276,7 @@ def create_predictor(args, mode, logger): min_input_shape = {"x": [1, 3, imgH, 10]} max_input_shape = {"x": [args.rec_batch_num, 3, imgH, 2304]} opt_input_shape = {"x": [args.rec_batch_num, 3, imgH, 320]} + config.exp_disable_tensorrt_ops(["transpose2"]) elif mode == "cls": min_input_shape = {"x": [1, 3, 48, 10]} max_input_shape = {"x": [args.rec_batch_num, 3, 48, 1024]} -- GitLab