diff --git a/tools/infer/utility.py b/tools/infer/utility.py index 3987d645718be83019cc84d99186c632405f5489..366212f228eec33f11c825bfaf1e360258af9b2e 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":