Skip to content

GitLab

Commit c8cc0fb4 authored by LDOUBLEV's avatar LDOUBLEV
Browse files

pre-commit

parent 0fd13fa4
Show whitespace changes
Inline Side-by-side
Showing with 2 additions and 2 deletions
tools/infer/utility.py
View file @ c8cc0fb4
...@@ -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% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment