minsubgraph_size 3 to 10

This commit is contained in:
LDOUBLEV 2021-07-06 14:49:38 +08:00
parent e4a939cbbc
commit 10e3729fd5
1 changed files with 1 additions and 1 deletions

View File

@ -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)