forgetest5
/
PaddleOCR
1
0
Fork 0
Code Issues Pull Requests Releases Wiki Activity

add parameter enable_mkldnn

This commit is contained in:
dyning 2020-07-14 14:23:28 +08:00
parent 7f8c64dfb0 fbdf80dc13
commit df79294cba
2 changed files with 6 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

7
tools/infer/utility.py
View File

@ -64,6 +64,7 @@ def parse_args():
type=str,
default="./ppocr/utils/ppocr_keys_v1.txt")
parser.add_argument("--use_space_char", type=bool, default=True)
parser.add_argument("--enable_mkldnn", type=bool, default=False)
return parser.parse_args()
@ -91,8 +92,10 @@ def create_predictor(args, mode):
config.enable_use_gpu(args.gpu_mem, 0)
else:
config.disable_gpu()
# config.enable_mkldnn()
config.set_cpu_math_library_num_threads(4)
config.set_cpu_math_library_num_threads(6)
if args.enable_mkldnn:
config.enable_mkldnn()
#config.enable_memory_optim()
config.disable_glog_info()