成功解决 RuntimeError:expected scalar type Double but found Float

该错误是由于数据的类型不对导致的,
但是不要被提示误导,并不是需要转换为double
而应该转为float32
假设数据变量为data
则将其转为float32代码如下

data = data.to(torch.float32)

这样就可以了

Guess you like

Origin blog.csdn.net/JingpengSun/article/details/131374565