fix_celoss
parent
e360fd6b71
commit
eb44b8dfd7
ppcls/loss
|
@ -47,7 +47,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…
Reference in New Issue