报错:ValueError_ Expected more than 1 value per channel when training, got input size torch.Size

除以batch_size后刚好余1,就发生了这个报错
数据加载器添加参数drop_last=True,把最后一个批次丢弃

self.train_loader = DataLoader(train_set, batch_size=10, shuffle=True, num_workers=2, drop_last=True)
  • 2
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 打赏
    打赏
  • 0
    评论
该错误通常表示在训练过程中,输入数据的通道数不满足模型的要求。你提供的错误信息中的torch.Size([1, 64, 1, 1])表示输入数据的维度是1个样本,每个样本有64个通道,每个通道的大小是1x1。模型期望每个通道至少有多于一个的值。 解决这个问题的方法是检查你的输入数据和模型的结构是否匹配。可能需要调整数据的通道数或者调整模型的输入层。 引用和引用中提到了相同的错误信息,但没有提供具体的解决方法。因此,我建议你查看引用中的解决方式总结,其中可能提供了解决这个问题的具体方法。<span class="em">1</span><span class="em">2</span><span class="em">3</span> #### 引用[.reference_title] - *1* [ValueError: Expected more than 1 value per channel when training, got input size torch.Size([1,](https://blog.csdn.net/weixin_47725177/article/details/128460187)[target="_blank" data-report-click={"spm":"1018.2226.3001.9630","extra":{"utm_source":"vip_chatgpt_common_search_pc_result","utm_medium":"distribute.pc_search_result.none-task-cask-2~all~insert_cask~default-1-null.142^v92^chatsearchT0_1"}}] [.reference_item style="max-width: 50%"] - *2* *3* [python报错系列(7)--ValueError: Expected more than 1 value perchannel when training, got input size ....](https://blog.csdn.net/qq_45365214/article/details/122670591)[target="_blank" data-report-click={"spm":"1018.2226.3001.9630","extra":{"utm_source":"vip_chatgpt_common_search_pc_result","utm_medium":"distribute.pc_search_result.none-task-cask-2~all~insert_cask~default-1-null.142^v92^chatsearchT0_1"}}] [.reference_item style="max-width: 50%"] [ .reference_list ]

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包

打赏作者

饭碗、碗碗香

感谢壮士的慷概解囊!

¥1 ¥2 ¥4 ¥6 ¥10 ¥20
扫码支付:¥1
获取中
扫码支付

您的余额不足,请更换扫码支付或充值

打赏作者

实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

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

余额充值