mirror of
https://github.com/PaddlePaddle/PaddleOCR.git
synced 2025-06-26 21:24:27 +00:00
Explicitly setting MKL-DNN (#15790)
This commit is contained in:
parent
1889d333a7
commit
86076a3746
@ -89,6 +89,7 @@ def prepare_common_init_args(model_name, common_args):
|
||||
elif device_type == "cpu":
|
||||
enable_mkldnn = common_args["enable_mkldnn"]
|
||||
if enable_mkldnn:
|
||||
pp_option.run_mode = "mkldnn"
|
||||
pp_option.mkldnn_cache_capacity = common_args["mkldnn_cache_capacity"]
|
||||
else:
|
||||
pp_option.run_mode = "paddle"
|
||||
|
Loading…
x
Reference in New Issue
Block a user