From ca8de0c69e7f4640e11052279d3e197f1eaaa5c0 Mon Sep 17 00:00:00 2001
From: xiaoting <31891223+tink2123@users.noreply.github.com>
Date: Tue, 20 Dec 2022 13:58:57 +0800
Subject: [PATCH] fix aster loss for axis (#8675)

---
 ppocr/losses/rec_aster_loss.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/ppocr/losses/rec_aster_loss.py b/ppocr/losses/rec_aster_loss.py
index 52605e46..9b0a34ee 100644
--- 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)
-- 
GitLab