mirrors/icefall
1
0
Fork 0
mirror of https://github.com/k2-fsa/icefall.git synced 2025-08-09 10:02:22 +00:00
Code Issues Packages Projects Releases Wiki Activity

fix export RNNLM onnx model typo (#1029)

Browse Source
This commit is contained in:
PF Luo 2023-04-28 19:53:06 +08:00 committed by GitHub
parent b0228c536e
commit 61ec3a7a8f
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
icefall/rnn_lm/export-onnx.py
View File

@ -232,7 +232,7 @@ def export_with_state(
L = 20
num_layers = model.rnn.num_layers
hidden_size = model.rnn.hidden_size
embedding_dim = model.rnn.embedding_dim
embedding_dim = model.embedding_dim
x = torch.randint(low=1, high=params.vocab_size, size=(N, L), dtype=torch.int64)
y = torch.randint(low=1, high=params.vocab_size, size=(N, L), dtype=torch.int64)