minsubgraph_size 3 to 10

10e3729f · LDOUBLEV · e4a939cb · 10e3729f
Commit 10e3729f authored Jul 06, 2021 by LDOUBLEV
Hide whitespace changes
Inline Side-by-side

Showing with 1 addition and 1 deletion

tools/infer/utility.py tools/infer/utility.py +1 -1

No files found.
--- a/tools/infer/utility.py
+++ b/tools/infer/utility.py
@@ -37,7 +37,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=3)
+    parser.add_argument("--min_subgraph_size", type=int, default=10)
    parser.add_argument("--precision", type=str, default="fp32")
    parser.add_argument("--gpu_mem", type=int, default=500)