From 96eb6287f77960f0a91a9e71bebe037638e485c4 Mon Sep 17 00:00:00 2001 From: littletomatodonkey Date: Tue, 24 May 2022 15:34:24 +0800 Subject: [PATCH] fix trt int8 inference (#6387) --- tools/infer/utility.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/tools/infer/utility.py b/tools/infer/utility.py index 3987d645..366212f2 100644 --- 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": -- GitLab