From 4fbd8f6fb61e1b4b48719aba3717423e8709298b Mon Sep 17 00:00:00 2001 From: FlyingQianMM <245467267@qq.com> Date: Wed, 2 Sep 2020 10:06:25 +0800 Subject: [PATCH] set use_mkl to True in deploy.py --- paddlex/deploy.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/paddlex/deploy.py b/paddlex/deploy.py index 747cf16..f69d8ee 100644 --- a/paddlex/deploy.py +++ b/paddlex/deploy.py @@ -30,7 +30,7 @@ class Predictor: model_dir, use_gpu=True, gpu_id=0, - use_mkl=False, + use_mkl=True, mkl_thread_num=4, use_trt=False, use_glog=False, @@ -94,7 +94,7 @@ class Predictor: use_gpu=True, gpu_id=0, use_mkl=False, - mkl_thread_num=mp.cpu_count(), + mkl_thread_num=4, use_trt=False, use_glog=False, memory_optimize=True): -- GitLab