diff --git a/tools/infer/utility.py b/tools/infer/utility.py index 74ec42ec842abe0f214f13eea6b30a613cfc517b..48b16db4a0f2c2c901509d691088d3dc4381fabd 100644 --- a/tools/infer/utility.py +++ b/tools/infer/utility.py @@ -275,6 +275,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]}