5% Speed up
This commit is contained in:
@@ -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')
|
||||
|
||||
Reference in New Issue
Block a user