Merge pull request #2720 from tink2123/for_paddle2.1

update for paddle2.1

Merge pull request #2720 from tink2123/for_paddle2.1
update for paddle2.1
a7f4c12f · xiaoting · GitHub · ac2e0da2 · c0d38801 · a7f4c12f
隐藏空白更改
内联并排

Showing with 1 addition and 3 deletion

ppocr/modeling/heads/self_attention.py ppocr/modeling/heads/self_attention.py +1 -3

未找到文件。
--- a/ppocr/modeling/heads/self_attention.py
+++ b/ppocr/modeling/heads/self_attention.py
@@ -319,9 +319,7 @@ class PrepareEncoder(nn.Layer):
        self.src_emb_dim = src_emb_dim
        self.src_max_len = src_max_len
        self.emb = paddle.nn.Embedding(
-            num_embeddings=self.src_max_len,
+            num_embeddings=self.src_max_len, embedding_dim=self.src_emb_dim)
-            embedding_dim=self.src_emb_dim,
-            sparse=True)
        self.dropout_rate = dropout_rate
    def forward(self, src_word, src_pos):