RuntimeError: Could not infer dtype of numpy.float32 问题解决

最近在跑深度学习的模型，训练模型和测试模型的时候都碰到了这个问题，最后发现问题出在了数据的处理上。

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")

发现可以成功训练模型啦！