Mirrors/modelscope
1
0
Fork 0
mirror of https://github.com/modelscope/modelscope.git synced 2026-09-01 19:49:03 +02:00
Code Issues Packages Projects Releases Wiki Activity

fix provider bug

Browse Source
This commit is contained in:
mulin.lyh
2023-09-28 17:14:42 +08:00
parent 146e69e6ba
commit c3e2a0e842
1 changed files with 4 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

8
tests/export/test_export_speech_signal_process.py
View File

@@ -60,13 +60,13 @@ class ExportSpeechSignalProcessTest(unittest.TestCase):
return
onnx_model = onnx.load(output)
onnx.checker.check_model(onnx_model)
ort_session = ort.InferenceSession(output)
ort_session = ort.InferenceSession(
output,
providers=['CUDAExecutionProvider', 'CPUExecutionProvider'])
with torch.no_grad():
model.eval()
outputs_origin = model.forward(dummy_inputs)
outputs_origin = numpify_tensor_nested(
outputs_origin,
providers=['CUDAExecutionProvider', 'CPUExecutionProvider'])
outputs_origin = numpify_tensor_nested(outputs_origin)
input_feed = {INPUT_NAME: dummy_inputs.numpy()}
outputs = ort_session.run(