最近在跑深度学习的模型,训练模型和测试模型的时候都碰到了这个问题,最后发现问题出在了数据的处理上。
pytorch版本:1.7.1
# 报错代码
label = label.astype("float32")
image = image.astype("float32")
查阅了许多博客后,发现了这篇文章:RUNTIMEERROR: COULD NOT INFER DTYPE OF COMPLEX (PYTORCH暂不支持复数数据类型),里面参考了pytorch官方文档的数据类型。
其中32位float类型可以用 torch.float32 或 torch.float 表示,于是将代码改为了以下:
# 修改代码
label = label.astype("float")
image = image.astype("float")
发现可以成功训练模型啦!