Parcourir la source

Update deploy.py

Jason il y a 5 ans
Parent
commit
41ffe47456
1 fichiers modifiés avec 1 ajouts et 1 suppressions
  1. 1 1
      paddlex/deploy.py

+ 1 - 1
paddlex/deploy.py

@@ -94,7 +94,7 @@ class Predictor:
                          use_gpu=True,
                          gpu_id=0,
                          use_mkl=False,
-                         mkl_thread_num=psutil.cpu_count(),
+                         mkl_thread_num=4,
                          use_trt=False,
                          use_glog=False,
                          memory_optimize=True):