From f6405ed4f7005013a031f5c88ed1b62635d4f118 Mon Sep 17 00:00:00 2001 From: megemini Date: Fri, 3 Jan 2025 15:55:46 +0800 Subject: [PATCH] [Fix] emb init --- paddlespeech/s2t/modules/align.py | 9 +++++++-- 1 file changed, 7 insertions(+), 2 deletions(-) diff --git a/paddlespeech/s2t/modules/align.py b/paddlespeech/s2t/modules/align.py index 34d796145..46d8796ab 100644 --- a/paddlespeech/s2t/modules/align.py +++ b/paddlespeech/s2t/modules/align.py @@ -69,8 +69,13 @@ class Embedding(nn.Embedding): name=None): if weight_attr is None: weight_attr = paddle.ParamAttr(initializer=nn.initializer.Normal()) - super(Embedding, self).__init__(num_embeddings, embedding_dim, - padding_idx, sparse, weight_attr, name) + super(Embedding, self).__init__( + num_embeddings=num_embeddings, + embedding_dim=embedding_dim, + padding_idx=padding_idx, + sparse=sparse, + weight_attr=weight_attr, + name=name) class Linear(nn.Linear):