diff --git a/ppocr/losses/basic_loss.py b/ppocr/losses/basic_loss.py index 4f9a9133adb39dbe7153b2c613554528685adbea..fa3ceda1b747aad3c4b275611b1257bf6950f013 100644 --- a/ppocr/losses/basic_loss.py +++ b/ppocr/losses/basic_loss.py @@ -79,7 +79,7 @@ class DMLLoss(nn.Layer): log_out2 = paddle.log(out2) loss = (F.kl_div( log_out1, out2, reduction='batchmean') + F.kl_div( - log_out2, log_out1, reduction='batchmean')) / 2.0 + log_out2, out1, reduction='batchmean')) / 2.0 return loss