diff --git a/refacer.py b/refacer.py index 8f54eb0..b241b82 100644 --- a/refacer.py +++ b/refacer.py @@ -61,7 +61,7 @@ class Refacer: print(f"CoreML mode with providers {self.providers}") elif 'CUDAExecutionProvider' in self.providers: self.mode = RefacerMode.CUDA - self.use_num_cpus = 1 + self.use_num_cpus = 2 self.sess_options.intra_op_num_threads = 1 if 'TensorrtExecutionProvider' in self.providers: self.providers.remove('TensorrtExecutionProvider')