expected scalar type Double but found Float

RuntimeError: expected scalar type Double but found Float原因：tensor的数据类型不正确解决：将数据类型转为float32并不是因为他expected Double你就要弄成Double查了一些博文说tensor.to(torch.float32)但其实这里面的tensor指的是tensor类型的数据在你自己的代码上他可能是x可能是da

不许人间见秃头

54551人浏览 · 2021-01-08 11:01:11

不许人间见秃头 · 2021-01-08 11:01:11 发布

RuntimeError: expected scalar type Double but found Float

原因：tensor的数据类型不正确
解决：将数据类型转为float32
并不是因为他expected Double你就要弄成Double
查了一些博文说
tensor.to(torch.float32)
但其实这里面的tensor指的是tensor类型的数据
在你自己的代码上他可能是x可能是data
这个要你自己多尝试
我的是x.to(torch.float32)
torch里有一些Bug
还有包括target后面要接.long()
分享出来希望帮到大家