- 错误信息
类型错误, 计算loss值的函数传入的参数类型不统一。 - 解决方法
查看上文loss计算代码部分的参数类型,如loss=f.mse_loss(out,label),检查out和label的类型都是torch.float类型即可。使用label.dtype查看tensor的类型。
PyTorch中由loss.backward()出现错误“RuntimeError: Found dtype Double but expected Float”
最新推荐文章于 2024-09-29 21:29:20 发布