소스 검색

set use_mkl to True in deploy.py

FlyingQianMM 5 년 전
부모
커밋
4fbd8f6fb6
1개의 변경된 파일2개의 추가작업 그리고 2개의 파일을 삭제
  1. 2 2
      paddlex/deploy.py

+ 2 - 2
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):