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

set use_mkl to True in deploy.py

上级 9746160e
......@@ -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):
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册