pytorch学习006- -回归问题的损失函数

L1Loss

from torch import nn
loss = nn.L1Loss()
  • 首先计算对应位置差值
  • 然后求绝对值
  • 累加
  • 除以总的像素数

MSELoss

from torch import nn
loss = nn.MSELoss()
  • 首先计算对应位置差值
  • 然后求平方值
  • 累加
  • 除以总的像素数

SmoothL1Loss

from torch import nn
loss = nn.SmoothL1Loss()
  • 首先计算对应位置差值
  • 如果差值不小1，则求绝对值后-0.5
  • 如果差值小于1，则求平方值后*0.5
  • 累加
  • 除以总的像素数
posted @ 2022-03-06 16:19  Keep_Silent  阅读(11)  评论(0)  编辑  收藏  举报