diff --git a/ppocr/losses/rec_aster_loss.py b/ppocr/losses/rec_aster_loss.py index 52605e46db35339cc22f7f1e6642456bfaf02f11..9b0a34eeac57089ae1d45ad9d8c0427b234c50c9 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)