fix aster loss for axis (#8674)

2022-12-20 13:22:03 +08:00 · 2022-12-20 13:22:03 +08:00 · 23e034c40e
parent b28af5d865
commit 23e034c40e
1 changed files with 1 additions and 1 deletions
--- a/ppocr/losses/rec_aster_loss.py
+++ b/ppocr/losses/rec_aster_loss.py
@ -28,7 +28,7 @@ class CosineEmbeddingLoss(nn.Layer):

    def forward(self, x1, x2, target):
        similarity = paddle.sum(
-            x1 * x2, dim=-1) / (paddle.norm(
+            x1 * x2, axis=-1) / (paddle.norm(
                x1, axis=-1) * paddle.norm(
                    x2, axis=-1) + self.epsilon)
        one_list = paddle.full_like(target, fill_value=1)