来自 Pytorch 的 MSELoss

Question

我正在尝试使用 Pytorch 训练神经网络。我希望损失函数是 MSE。我尝试使用 torch.nn.MSELoss，但出现了我不理解的错误。

例如下面的代码给我 RuntimeError: Boolean value of Tensor with more than one value is ambiguous

import torch
import torch.nn as nn

model = torch.zeros(64)
model.requires_grad = True
target = torch.ones(64)

loss = nn.MSELoss(model, target)

非常感谢任何帮助！

Answer 1

请查看 Pytorch 文档：https://pytorch.org/docs/stable/generated/torch.nn.MSELoss.html

在调用目标和预测之前，您需要创建一个 MSELoss 对象。

loss = nn.MSELoss()
input = torch.zeros(64, requires_grad=True)
target = torch.ones(64)
output = loss(input, target)