From ab0b1ace8fe0cfd140db2a4b5b37ecf50c0ddfd2 Mon Sep 17 00:00:00 2001 From: LDOUBLEV Date: Thu, 10 Jun 2021 14:29:56 +0800 Subject: [PATCH] add params min_subgraph_size --- tools/infer/utility.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/tools/infer/utility.py b/tools/infer/utility.py index a87e3eefb2..dbdb4e3399 100755 --- a/tools/infer/utility.py +++ b/tools/infer/utility.py @@ -236,8 +236,8 @@ def create_predictor(args, mode, logger): config.enable_tensorrt_engine( precision_mode=inference.PrecisionType.Float32, max_batch_size=args.max_batch_size, - min_subgraph_size=args. - min_subgraph_size) # skip the minmum trt subgraph + min_subgraph_size=args.min_subgraph_size) + # skip the minmum trt subgraph if mode == "det": min_input_shape = { "x": [1, 3, 50, 50],