From e60d56a564d496fda9306bda653549ccb75bf34d Mon Sep 17 00:00:00 2001 From: "mulin.lyh" Date: Sun, 8 Oct 2023 15:18:31 +0800 Subject: [PATCH] fix test_export_speech_signal_process.py issue --- tests/export/test_export_speech_signal_process.py | 4 +--- 1 file changed, 1 insertion(+), 3 deletions(-) diff --git a/tests/export/test_export_speech_signal_process.py b/tests/export/test_export_speech_signal_process.py index faba59a5..d4579936 100644 --- a/tests/export/test_export_speech_signal_process.py +++ b/tests/export/test_export_speech_signal_process.py @@ -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(