fix_celoss
parent
3bd78eb1d3
commit
6714a9d1f9
|
@ -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…
Reference in New Issue