From 182bf94343132a0f33d25e79593b06949c67302d Mon Sep 17 00:00:00 2001 From: Bin Lu Date: Mon, 31 May 2021 14:28:52 +0800 Subject: [PATCH] Update celoss.py --- ppcls/losses/celoss.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) 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() +