前言
pytorch常用损失函数
损失函数的基本用法:
criterion = LossCriterion() #构造函数有自己的参数 loss = criterion(x, y) #调用标准时也有参数
得到的loss结果已经对mini-batch数量取了平均值
1.BCELoss(二分类)
CLASS torch.nn.BCELoss(weight=None, size_average=None, reduce=None, reduction='mean')
创建一个衡量目标和输出之间二进制交叉熵的criterion
unreduced loss函数(即reduction参数设置为'none')为:
N表示batch size,xn为输出,yn为目标