fix kldiv input

95d07675 · littletomatodonkey · e45ff483 · 95d07675
Commit 95d07675 authored Jun 09, 2021 by littletomatodonkey
Hide whitespace changes
Inline Side-by-side

Showing with 1 addition and 1 deletion

ppocr/losses/basic_loss.py ppocr/losses/basic_loss.py +1 -1

No files found.
--- 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