阅读背景:

loss.backward()处遇到“RuntimeError: Found dtype Double but expected Float”

来源:互联网 
错误信息

类型错误, 计算loss值的函数传入的参数类型不统一。

解决方法

查看上文loss计算代码部分的参数类型,如loss=f.mse_loss(out,label),检查out和label的类型都是torch.float类型即可。使用label.dtype查看tensor的类型。查看上文loss计算代




你的当前访问异常,请进行认证后继续阅读剩余内容。

分享到: