当调用 loss.backward() 函数时,出现以下错误 RuntimeError: expected dtype Double but got dtype Float (validate_dtype at …\aten\src\ATen\native\TensorIterator.cpp:143) 报错。
解决方法:
当你使用的损失函数是nn.MSELoss 时,要保证输入数据的类型是一样的。
如:nn.MSELoss(Pred,Result)。要保证Pred和Result的数据类型是一样。不能一个是float32,另外一个是float64.