Loss = criterion(prediction3, t_ytrain) + criterion(prediction4, t_yResidual_train)
原始代码如上所示,报错,说希望Float类型
修改之后的如下所示:
Loss = criterion(prediction3, t_ytrain) + criterion(prediction4, t_yResidual_trainto(torch.float32))
发现错误的操作:
在模型训练过程中Debug
一步一步操作,在计算Loss.backward()的时候跳入报警程序
这是因为损失函数中计算的类型不一致导致的。
希望我的方法对你们有帮助。