pytorch报错：ValueError: Expected more than 1 value per channel when training, got input size torch.Siz

pytorch报错：ValueError: Expected more than 1 value per channel when training, got input size torch.Size([1,256,1,1])通过查阅之后：解决方法1.在torch.utils.data.DataLoader类中或自己创建的继承于DataLoader的类中设置参数drop_last=True，把不

双羽Min

2018人浏览 · 2021-12-01 12:15:27

双羽Min · 2021-12-01 12:15:27 发布

pytorch报错：ValueError: Expected more than 1 value per channel when training, got input size torch.Size([1,256,1,1])

通过查阅之后：
解决方法1.在torch.utils.data.DataLoader类中或自己创建的继承于DataLoader的类中设置参数drop_last=True，把不够一个batch_size的数据丢弃。（实际操作不行）

解决方法2，batch_size大于等于2，我调节batch_size=4不报错

技术共进，成长同行——讯飞AI开发者社区

更多推荐

多模态RAG入门到精通：原理+流程全拆解，收藏这篇就够了！

讯飞AI开发者社区

V-JEPA 2：自监督视频模型可实现理解、预测和规划

讯飞AI开发者社区

Mem0 + Milvus：为人工智能构建持久化长时记忆

https://github.com/fdaf23123/7/issues/130https://github.com/fdaf23123/7/issues/131https://github.com/fdaf23123/7/issues/132https://github.com/fdaf23123/7/issues/133https://github.com/fdaf23123/7/issue