ignore_index:交叉熵损失中的ignore_index参数【torch.nn.functional.cross_entropy的ignore_index】【默认-100】

1. 交叉熵损失 CrossEntropyLoss

CrossEntropyLoss 交叉熵损失可函数以用于分类或者分割任务中,这里主要介绍分割任务

建立如下的数据,pred是预测样本,label是真实标签

分割中,使用交叉熵损失的话,需要保证label的维度比pred维度少1,也就是没有channel维度。并且,label的类型是int

正常计算损失结果为:

手动计算一下,pred的softmax

猜你喜欢

转载自blog.csdn.net/u013250861/article/details/131762635