最近在跑深度学习的模型,训练模型和测试模型的时候都碰到了这个问题,最后发现问题出在了数据的处理上。
pytorch版本:1.7.1
# 报错代码
label = label.astype("float32")
image = image.astype("float32")
查阅了许多博客后,发现了这篇文章:RUNTIMEERROR: COULD NOT INFER DTYPE OF COMPLEX (PYTORCH暂不支持复数数据类型),里面参考了pytorch官方文档的数据类型。