diff --git a/tools/infer/utility.py b/tools/infer/utility.py index 1c82280099..9a687b73f6 100755 --- a/tools/infer/utility.py +++ b/tools/infer/utility.py @@ -35,7 +35,7 @@ def init_args(): parser.add_argument("--use_gpu", type=str2bool, default=True) parser.add_argument("--ir_optim", type=str2bool, default=True) parser.add_argument("--use_tensorrt", type=str2bool, default=False) - parser.add_argument("--min_subgraph_size", type=int, default=10) + parser.add_argument("--min_subgraph_size", type=int, default=15) parser.add_argument("--precision", type=str, default="fp32") parser.add_argument("--gpu_mem", type=int, default=500)