fix trt int8 inference (#6387)

2022-05-24 15:34:24 +08:00 · 2022-05-24 15:34:24 +08:00 · 96eb6287f7
parent f2c0a1a2fd
commit 96eb6287f7
1 changed files with 2 additions and 1 deletions
--- a/tools/infer/utility.py
+++ b/tools/infer/utility.py
@ -202,7 +202,8 @@ def create_predictor(args, mode, logger):
                    workspace_size=1 << 30,
                    precision_mode=precision,
                    max_batch_size=args.max_batch_size,
-                    min_subgraph_size=args.min_subgraph_size)
+                    min_subgraph_size=args.min_subgraph_size,
+                    use_calib_mode=False)
                # skip the minmum trt subgraph
            use_dynamic_shape = True
            if mode == "det":