提交 c8cc0fb4 编写于 作者: L LDOUBLEV

pre-commit

上级 0fd13fa4
...@@ -224,8 +224,8 @@ def create_predictor(args, mode, logger): ...@@ -224,8 +224,8 @@ def create_predictor(args, mode, logger):
if hasattr(args, "cpu_threads"): if hasattr(args, "cpu_threads"):
config.set_cpu_math_library_num_threads(args.cpu_threads) config.set_cpu_math_library_num_threads(args.cpu_threads)
else: else:
config.set_cpu_math_library_num_threads( # default cpu threads as 10
10) # default cpu threads as 10 config.set_cpu_math_library_num_threads(10)
if args.enable_mkldnn: if args.enable_mkldnn:
# cache 10 different shapes for mkldnn to avoid memory leak # cache 10 different shapes for mkldnn to avoid memory leak
config.set_mkldnn_cache_capacity(10) config.set_mkldnn_cache_capacity(10)
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册