Я пытаюсь обучить нейронную сеть с помощью 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)
Любая помощь будет очень признательна!
Пожалуйста, посмотрите в документации 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)
Большое спасибо! Извините, я еще не очень хорошо знаком с pytorch.