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

fix test_export_speech_signal_process.py issue

Browse Source
This commit is contained in:
mulin.lyh
2023-10-08 15:18:31 +08:00
parent d0384a5849
commit e60d56a564
1 changed files with 1 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
tests/export/test_export_speech_signal_process.py
View File

@@ -66,9 +66,7 @@ class ExportSpeechSignalProcessTest(unittest.TestCase):
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(