fix aster loss for axis (PaddlePaddle#8675)

rainbow-jay · Dec 20, 2022 · ca8de0c · ca8de0c
1 parent 582e868
commit ca8de0c
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)