From bad44a7aed51003e6f945f06f896d9a601acb176 Mon Sep 17 00:00:00 2001
From: Yifan Yang <64255737+yfyeung@users.noreply.github.com>
Date: Fri, 5 Jul 2024 14:52:35 +0800
Subject: [PATCH] Update decoder.py

---
 egs/librispeech/ASR/zipformer_lstm/decoder.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/egs/librispeech/ASR/zipformer_lstm/decoder.py b/egs/librispeech/ASR/zipformer_lstm/decoder.py
index 4e690892b..51c1952b9 100644
--- a/egs/librispeech/ASR/zipformer_lstm/decoder.py
+++ b/egs/librispeech/ASR/zipformer_lstm/decoder.py
@@ -58,7 +58,7 @@ class Decoder(nn.Module):
 
         self.embedding = nn.Embedding(
             num_embeddings=vocab_size,
-            decoder_dim=decoder_dim,
+            embedding_dim=decoder_dim,
         )
         # the balancers are to avoid any drift in the magnitude of the
         # embeddings, which would interact badly with parameter averaging.