mirrors/PaddleClas
1
0
Fork 0
mirror of https://github.com/PaddlePaddle/PaddleClas.git synced 2025-06-03 21:55:06 +08:00
Code Issues Projects Releases Wiki Activity

Update celoss.py

Browse Source
This commit is contained in:
Bin Lu 2021-05-31 14:28:52 +08:00 committed by GitHub
parent 8edec09dd4
commit 182bf94343
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 2 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
ppcls/losses/celoss.py
View File

@ -111,4 +111,5 @@ class KLDivLoss(paddle.nn.Layer):
if is_logit: if is_logit:
p = paddle.nn.functional.softmax(p) p = paddle.nn.functional.softmax(p)
q = paddle.nn.functional.softmax(q) q = paddle.nn.functional.softmax(q)
return -(p * paddle.log(q + 1e-8)).sum(1).mean() return -(p * paddle.log(q + 1e-8)).sum(1).mean()