MSELoss(size_average=False)

Posted 2021-03-07 junzhaoliang

tags:

篇首语：本文由小常识网(cha138.com)小编为大家整理，主要介绍了MSELoss(size_average=False)相关的知识，希望对你有一定的参考价值。

nn.MSELoss()得到的是平均到像素的loss

nn.MSELoss(size_average=False)得到整个batch所有像素loss和

MSELoss(size_average=False).div(batch_size)得到平均图像loss

a=torch.Tensor([[1,1],[1,1]])
b=torch.Tensor([[0,0],[0,0]])

f = nn.MSELoss()
f(a,b)
>>>tensor(1.)

f = nn.MSELoss(size_average=False)
f(a,b)
>>>tensor(4.)

以上是关于MSELoss(size_average=False)的主要内容，如果未能解决你的问题，请参考以下文章

Pytorch MSELoss

神经网络架构pytorch－MSELoss损失函数

PyTorch学习笔记：nn.MSELoss——MSE损失

Pytorch_torch.nn.MSELoss

Pytorch常用损失函数nn.BCEloss()；nn.BCEWithLogitsLoss()；nn.CrossEntropyLoss()；nn.L1Loss()； nn.MSELoss()；(代码