mirrors/icefall
1
0
Fork 0
mirror of https://github.com/k2-fsa/icefall.git synced 2025-09-19 05:54:20 +00:00
Code Issues Packages Projects Releases Wiki Activity

fix CI tests

Browse Source
This commit is contained in:
Fangjun Kuang 2023-07-25 13:16:35 +08:00
parent 2ce61e6e7f
commit 721c8e95a5
1 changed files with 2 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
egs/librispeech/ASR/pruned_transducer_stateless7/test_onnx.py
View File

@ -265,7 +265,7 @@ def test_zipformer_encoder():
torch.onnx.export(
encoder,
(x),
(x, torch.ones(1, dtype=torch.float32)),
filename,
verbose=False,
opset_version=opset_version,
@ -289,6 +289,7 @@ def test_zipformer_encoder():
input_nodes = session.get_inputs()
inputs = {
input_nodes[0].name: x.numpy(),
input_nodes[1].name: torch.ones(1, dtype=torch.float32).numpy(),
}
onnx_y = session.run(["y"], inputs)[0]
onnx_y = torch.from_numpy(onnx_y)