From 61ec3a7a8fc8be859b23a821e568950fb898b37a Mon Sep 17 00:00:00 2001 From: PF Luo Date: Fri, 28 Apr 2023 19:53:06 +0800 Subject: [PATCH] fix export RNNLM onnx model typo (#1029) --- icefall/rnn_lm/export-onnx.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/icefall/rnn_lm/export-onnx.py b/icefall/rnn_lm/export-onnx.py index b6d3a03ed..1d9af5e3d 100755 --- a/icefall/rnn_lm/export-onnx.py +++ b/icefall/rnn_lm/export-onnx.py @@ -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)