5% Speed up

This commit is contained in:
Andras Fekete
2023-06-19 10:50:58 -04:00
parent 1388fc651f
commit 8e9035f712

View File

@@ -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')