损失函数
损失函数:衡量模型输出与真实标签之间的差异。与损失函数有关的还有代价函数和目标函数
常见的损失函数
交叉熵损失 CrossEntropyLoss
torch.nn.CrossEntropyLoss(weight=None,
size_average=None,
ignore_index=- 100,
reduce=None,
reduction='mean', label_smoothing=0.0)
- weight:各类别的loss设置权值
- ignore_index:忽略某个类别
- reduction:计算模式,可为none/sum/mean,none 表示逐个元素计算,这样有多少个样本就会返回多少个 loss。sum 表示所有元素的 loss求和,返回标量,mean 所有元素的 loss 求