admin/LC_NeoRefacer
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

5% Speed up

Browse Source
This commit is contained in:
Andras Fekete
2023-06-19 10:50:58 -04:00
parent 1388fc651f
commit 8e9035f712
1 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
refacer.py
View File

@@ -61,7 +61,7 @@ class Refacer:
print(f"CoreML mode with providers {self.providers}") print(f"CoreML mode with providers {self.providers}")
elif 'CUDAExecutionProvider' in self.providers: elif 'CUDAExecutionProvider' in self.providers:
self.mode = RefacerMode.CUDA self.mode = RefacerMode.CUDA
self.use_num_cpus = 1 self.use_num_cpus = 2
self.sess_options.intra_op_num_threads = 1 self.sess_options.intra_op_num_threads = 1
if 'TensorrtExecutionProvider' in self.providers: if 'TensorrtExecutionProvider' in self.providers:
self.providers.remove('TensorrtExecutionProvider') self.providers.remove('TensorrtExecutionProvider')