mirror of
https://github.com/PaddlePaddle/PaddleClas.git
synced 2025-06-03 21:55:06 +08:00
Update celoss.py
This commit is contained in:
parent
e5771019e5
commit
2e41b5de5e
@ -51,7 +51,6 @@ class CELoss(nn.Layer):
|
||||
loss = paddle.sum(x * label, axis=-1)
|
||||
else:
|
||||
if label.shape[-1] == x.shape[-1]:
|
||||
label = F.softmax(label, axis=-1)
|
||||
soft_label = True
|
||||
else:
|
||||
soft_label = False
|
||||
|
Loading…
x
Reference in New Issue
Block a user