diff --git a/tests/export/test_export_speech_signal_process.py b/tests/export/test_export_speech_signal_process.py index 4c6a0067..d4579936 100644 --- a/tests/export/test_export_speech_signal_process.py +++ b/tests/export/test_export_speech_signal_process.py @@ -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(