nn.L1Loss
取预测值和真实值的绝对误差的平均数。
loss(x,y)=1N∑i=1N∣x−y∣ \operatorname{loss}(\mathbf{x}, \mathbf{y})=\frac{1}{N} \sum_{i=1}^{N}|\mathbf{x}-\mathbf{y}| loss(x,y)=N1i=1∑N∣x−y∣
nn.SmoothL1Loss
也叫作 Huber Loss,误差在 (-1,1) 上是平方损失,其他情况是 L1 损失。
loss(x,y)=1N{
12(xi−yi)2 if ∣xi−yi∣<1∣xi−yi∣−12, otherwise \operatorname{loss}(\mathbf{x}, \mathbf{y})=\frac{1}{N}\left\{\begin{array}{cc}{\frac{1}{2}\left(\mathbf{x}_{i}-\mathbf{y}_{i}\right)^{2}} & {\text { if }\left|\mathbf{x}_{i}-\mathbf{y}_{i}\right|<1} \\ {\left|\mathbf{x}_{i}-\mathbf{y}_{i}\right|-\frac{1}{2},} & {\text { otherwise }}\end{array}\right. loss(x,y)=N1{
21(xi−y

本文总结了Torch中常用的损失函数,包括L1Loss、SmoothL1Loss(Huber Loss)、MSELoss(均方误差)、BCELoss(二值交叉熵)和CrossEntropyLoss。这些损失函数在不同场景下有不同的应用,如L1Loss和MSELoss常用于回归任务,BCELoss适合二分类,而CrossEntropyLoss则适用于多分类问题。
最低0.47元/天 解锁文章
8445





