[Fix] emb init (#3962)

9 months ago · 553a9db374
parent 7d26f93d2c
commit 553a9db374
1 changed files with 7 additions and 2 deletions
--- 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):