Ver código fonte

Update deploy.py

Jason 5 anos atrás
pai
commit
41ffe47456
1 arquivos alterados com 1 adições e 1 exclusões
  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):