如何修复RuntimeError“标量类型为Float的预期对象，但参数为标量类型Double”？

如何修复RuntimeError“标量类型为Float的预期对象，但参数为标量类型Double”？

参考来自这个github问题。

当错误出现时

RuntimeError: Expected object of scalar type Float but got scalar typeDouble for argument #4 'mat1'

，您将需要使用该

.float()

函数，因为它说的是

Expected object ofscalar type Float

。

因此，解决方案更改

y_pred = model(X_trainTensor)

为

y_pred =model(X_trainTensor.float())

。

同样，当您遇到另一个错误时

loss = loss_fn(y_pred,y_trainTensor)

，

y_trainTensor.long()

由于错误消息中显示，因此您也需要

Expected object ofscalar type Long

。

您也可以

model.double()

按照@Paddy的建议进行 *** 作。