pytorch挖坑之:Expected object of scalar type Long but got scalar type Float for argument target...

本文记录了一次在PyTorch中遇到的类型转换错误,原本期望的是Long类型,但实际接收到的是Float类型。经过排查,问题出现在自定义的Dataset类中,对标签进行了不必要的one-hot编码。正确的做法是保持标签为一维,即使输出层有多个维度。
摘要由CSDN通过智能技术生成

1. 问题分析

报错提示:需要Long类型的,但是收到了Float类型。

google了一下,都说是到相应的位置,后面加上.float()。

但是改了之后,还是不行,报了另外一个错误:
在这里插入图片描述后来发现,应该是自己传的target有问题,也就是label。

查看了一下自己写的Dataset类,手动地把0变成[1, 0], 把1变成[0, 1],
在这里插入图片描述
**其实根本不需要!!!不需要!!!不需要!!!

虽然设置的输出有两个维度,但是这里的label就是1-D的就可以了。修改后,如下:
在这里插入图片描述

2. 总结

没啥好总结的。再见

结束。

评论 2
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值