提交 4fbd8f6f 编写于 作者: F FlyingQianMM

set use_mkl to True in deploy.py

上级 9746160e
...@@ -30,7 +30,7 @@ class Predictor: ...@@ -30,7 +30,7 @@ class Predictor:
model_dir, model_dir,
use_gpu=True, use_gpu=True,
gpu_id=0, gpu_id=0,
use_mkl=False, use_mkl=True,
mkl_thread_num=4, mkl_thread_num=4,
use_trt=False, use_trt=False,
use_glog=False, use_glog=False,
...@@ -94,7 +94,7 @@ class Predictor: ...@@ -94,7 +94,7 @@ class Predictor:
use_gpu=True, use_gpu=True,
gpu_id=0, gpu_id=0,
use_mkl=False, use_mkl=False,
mkl_thread_num=mp.cpu_count(), mkl_thread_num=4,
use_trt=False, use_trt=False,
use_glog=False, use_glog=False,
memory_optimize=True): memory_optimize=True):
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册