From b9d243f854eadee092bb780821cf8db6d1703090 Mon Sep 17 00:00:00 2001 From: littletomatodonkey <2120160898@bit.edu.cn> Date: Fri, 11 Dec 2020 00:24:14 +0800 Subject: [PATCH] Update utils.py --- tools/infer/utils.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/tools/infer/utils.py b/tools/infer/utils.py index f31fca914..2923bcf0c 100644 --- a/tools/infer/utils.py +++ b/tools/infer/utils.py @@ -43,7 +43,7 @@ def parse_args(): parser.add_argument("--gpu_mem", type=int, default=8000) parser.add_argument("--enable_benchmark", type=str2bool, default=False) parser.add_argument("--top_k", type=int, default=1) - parser.add_argument("--enable_mkldnn", type=bool, default=False) + parser.add_argument("--enable_mkldnn", type=str2bool, default=False) parser.add_argument("--hubserving", type=str2bool, default=False) # params for infer