diff --git a/ppcls/losses/celoss.py b/ppcls/losses/celoss.py index e588bb8f6..69f6e1177 100644 --- a/ppcls/losses/celoss.py +++ b/ppcls/losses/celoss.py @@ -111,4 +111,5 @@ class KLDivLoss(paddle.nn.Layer): if is_logit: p = paddle.nn.functional.softmax(p) q = paddle.nn.functional.softmax(q) - return -(p * paddle.log(q + 1e-8)).sum(1).mean() \ No newline at end of file + return -(p * paddle.log(q + 1e-8)).sum(1).mean() +