set explicit execution provider (#36)

rwth-i6 · Sep 22, 2023 · aaa51af · aaa51af
1 parent f6fdaf7
commit aaa51af
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/tests/test_blstm.py b/tests/test_blstm.py
@@ -33,7 +33,7 @@ def test_blstm_onnx_export():
                 "classes": {0: "batch", 1: "time"},
             },
         )
-        session = ort.InferenceSession(f.name)
+        session = ort.InferenceSession(f.name, providers=["CPUExecutionProvider"])
         outputs_onnx = torch.FloatTensor(
             session.run(None, {"data": dummy_data.numpy(), "data_len": dummy_data_len.numpy()})[0]
         )