adjust the min_subgraph_size to 15

This commit is contained in:
LDOUBLEV 2021-08-26 02:21:22 +00:00
parent 71ed68c9a8
commit d7cbe52697
1 changed files with 1 additions and 1 deletions

View File

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