RuntimeError: Found dtype Double but expected Float”
created at 10-28-2021
views: 4
An error was reported when calculating the loss function,
Solution¶
target.float()
a=np.array([[1,2],[3,4]])
b=np.array([[2,3],[4,4]])
loss_fn = torch.nn.MSELoss(reduce=True, size_average=True)
input = torch.autograd.Variable(torch.from_numpy(a))
target = torch.autograd.Variable(torch.from_numpy(b))
loss = loss_fn(input.float(), target.float())
print(loss)
