fix trt int8 inference (#6387)

96eb6287 · littletomatodonkey · GitHub · f2c0a1a2 · 96eb6287
显示空白变更内容
内联并排

Showing with 2 addition and 1 deletion

tools/infer/utility.py tools/infer/utility.py +2 -1

未找到文件。
--- 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":