Skip to content

Commit 61ec3a7

Browse files
authored
fix export RNNLM onnx model typo (#1029)
1 parent b0228c5 commit 61ec3a7

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

icefall/rnn_lm/export-onnx.py

+1-1
Original file line numberDiff line numberDiff line change
@@ -232,7 +232,7 @@ def export_with_state(
232232
L = 20
233233
num_layers = model.rnn.num_layers
234234
hidden_size = model.rnn.hidden_size
235-
embedding_dim = model.rnn.embedding_dim
235+
embedding_dim = model.embedding_dim
236236

237237
x = torch.randint(low=1, high=params.vocab_size, size=(N, L), dtype=torch.int64)
238238
y = torch.randint(low=1, high=params.vocab_size, size=(N, L), dtype=torch.int64)

0 commit comments

Comments
 (0)