[Fix] emb init (#3962)

pull/3964/head
megemini 9 months ago committed by GitHub
parent 7d26f93d2c
commit 553a9db374
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194

@ -69,8 +69,13 @@ class Embedding(nn.Embedding):
name=None): name=None):
if weight_attr is None: if weight_attr is None:
weight_attr = paddle.ParamAttr(initializer=nn.initializer.Normal()) weight_attr = paddle.ParamAttr(initializer=nn.initializer.Normal())
super(Embedding, self).__init__(num_embeddings, embedding_dim, super(Embedding, self).__init__(
padding_idx, sparse, weight_attr, name) num_embeddings=num_embeddings,
embedding_dim=embedding_dim,
padding_idx=padding_idx,
sparse=sparse,
weight_attr=weight_attr,
name=name)
class Linear(nn.Linear): class Linear(nn.Linear):

Loading…
Cancel
Save