diff --git a/tools/infer/utility.py b/tools/infer/utility.py index de3bb3db9742859ed2f098d90c87c1610e0d7cfe..ff4a0276e2dc690faaa5c0f22a4c88a9f31a75f9 100755 --- a/tools/infer/utility.py +++ b/tools/infer/utility.py @@ -224,8 +224,8 @@ def create_predictor(args, mode, logger): if hasattr(args, "cpu_threads"): config.set_cpu_math_library_num_threads(args.cpu_threads) else: - config.set_cpu_math_library_num_threads( - 10) # default cpu threads as 10 + # default cpu threads as 10 + config.set_cpu_math_library_num_threads(10) if args.enable_mkldnn: # cache 10 different shapes for mkldnn to avoid memory leak config.set_mkldnn_cache_capacity(10)