Bootstrap

运行深度学习模型出现“RuntimeError:expected scalar type Long but found Float”, 成功解决!

最近在做深度学习分类时遇到了如标题所示的错误,奈何不知如何修改,最后经过探索,成功解决!
下面直接上报错问题及解决方案。

错误在这里插入图片描述

在这里插入图片描述

解决方案

实际做分类时:分类的标签应该是long,图像是float32
因此,将数据类型修改一下就成功了!
在这里插入图片描述

成功解决上述问题!

;