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

fix_celoss

Browse Source
This commit is contained in:
cuicheng01 2022-11-14 13:00:28 +00:00 committed by cuicheng01
parent e360fd6b71
commit eb44b8dfd7
1 changed files with 0 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

1
ppcls/loss/celoss.py
View File

@ -47,7 +47,6 @@ class CELoss(nn.Layer):
loss = paddle.sum(x * label, axis=-1) loss = paddle.sum(x * label, axis=-1)
else: else:
if label.shape[-1] == x.shape[-1]: if label.shape[-1] == x.shape[-1]:
label = F.softmax(label, axis=-1)
soft_label = True soft_label = True
else: else:
soft_label = False soft_label = False