Disabled TensorrtExecutionProvider due to a bug, it will now run with CUDAExecutionProvider on NVIDIA until we resolve error #9 #10
This commit is contained in:
@@ -56,6 +56,8 @@ class Refacer:
|
||||
self.mode = RefacerMode.CUDA
|
||||
self.use_num_cpus = 1
|
||||
self.sess_options.intra_op_num_threads = 1
|
||||
if 'TensorrtExecutionProvider' in self.providers:
|
||||
self.providers.remove('TensorrtExecutionProvider')
|
||||
print(f"CUDA mode with providers {self.providers}")
|
||||
"""
|
||||
elif 'TensorrtExecutionProvider' in self.providers:
|
||||
|
||||
Reference in New Issue
Block a user