ValueError: Expected more than 1 value per channel when training, got input size torch.Size([1, 384]

本文讨论了在使用PyTorch训练神经网络时遇到的输入数据维度不符预期的问题,特别是在数据集大小无法被批次大小整除时。解决方案是调整DataLoader类的drop_last参数为True,以处理不完整的批次。

摘要生成于 C知道 ,由 DeepSeek-R1 满血版支持, 前往体验 >

        这个错误通常发生在使用PyTorch训练神经网络时,输入数据维度不符合预期,不能batch_size整除。可能是输入的数据有剩余但不足以达到batch_size所导致的

查阅资料发现:在torch.utils.data这个包中,  DataLoader类下有一参数为:

drop_last – set to True to drop the last incomplete batch, if the dataset size is not divisible by the batch size. If False and the size of dataset is not divisible by the batch size, then the last batch will be smaller. (default: False)

drop_last-如果数据集大小不能被批次大小整除,则设置为True可删除最后一个不完整的批次。如果为False,并且数据集的大小不能被批次大小整除,则最后一个批次将更小。(默认值:False)

解决办法:只需要将这里的drop_last=True,问题便可解决。

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值