adjust the min_subgraph_size to 15

truongnn1106 · Aug 26, 2021 · d7cbe52 · d7cbe52
1 parent 71ed68c
commit d7cbe52
Showing 1 changed file with 1 addition and 1 deletion.
diff --git 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)