RuntimeError: expected scalar type Float but found Double

这个问题出现一般是tensor的数据类型不对,在训练过程中报错。

看具体问题出在哪里,有可能是网络层中数据类型没对上,也有可能是训练或者测试过程中。

解决办法:

1、数据一开始就定义错了

在设置festures的时候dtype设置的是float应该和下面一致是torch.float

2、在网络层出现错误

RuntimeError: expected scalar type Float but found Double_第1张图片

在对应位置添加.to(torch.float32)即可解决。

综上,该问题的出现即数据类型未对应好。

你可能感兴趣的:(python,深度学习,人工智能)