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

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

错误运行深度学习模型出现“RuntimeError:expected scalar type Long but found Float”, 成功解决!_第1张图片

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

解决方案

实际做分类时:分类的标签应该是long,图像是float32
因此,将数据类型修改一下就成功了!
运行深度学习模型出现“RuntimeError:expected scalar type Long but found Float”, 成功解决!_第3张图片

成功解决上述问题!

你可能感兴趣的:(pytorch深度学习,微波遥感,SAR与深度学习,bug,debug,深度学习,ResNet自制数据集分类)