fix aster loss for axis (#8674)

PaddlePaddle · Dec 20, 2022 · 23e034c · 23e034c
1 parent b28af5d
commit 23e034c
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/ppocr/losses/rec_aster_loss.py b/ppocr/losses/rec_aster_loss.py
@@ -28,7 +28,7 @@ def __init__(self, margin=0.):
 
     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)