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