Runtime error: expected scalar type Float but found Double

Runtime error: expected scalar type Float but found Double

Runtime error: expected scalar type Float but found Double

产生原因及解决方法

  • 产生原因 数据是floaat64,而torch默认是float32,于是报错

  • 解决方法:如下代码

torch.tensor(data,dtype=torch.float32)

以上就是全部内容

你可能感兴趣的:(pytorch,python,深度学习,开发语言)