Merge pull request #6285 from littletomatodonkey/tmp

fix rec prediction for trt7
2026-09-24 23:33:08 +08:00 · 2022-05-16 12:05:48 +08:00
parent bb77fcef6f 265cd32bd5
commit 0571a41db1
1 changed files with 1 additions and 0 deletions
@@ -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]}