pytorch挖坑之:Expected object of scalar type Long but got scalar type Float for argument target...
最新推荐文章于 2023-07-04 15:40:39 发布
本文记录了一次在PyTorch中遇到的类型转换错误,原本期望的是Long类型,但实际接收到的是Float类型。经过排查,问题出现在自定义的Dataset类中,对标签进行了不必要的one-hot编码。正确的做法是保持标签为一维,即使输出层有多个维度。
摘要由CSDN通过智能技术生成