0.1.4.post3
This commit is contained in:
@@ -58,7 +58,7 @@ import numpy as np
|
||||
if 'CUDAExecutionProvider' in ort.get_available_providers():
|
||||
provider = 'CUDAExecutionProvider'
|
||||
else:
|
||||
provider = 'CPUExecutionProvider"
|
||||
provider = 'CPUExecutionProvider'
|
||||
opts = ort.SessionOptions()
|
||||
opts.intra_op_num_threads = 16
|
||||
opts.execution_mode = ort.ExecutionMode.ORT_SEQUENTIAL
|
||||
|
||||
Reference in New Issue
Block a user