ValueError: Expected more than 1 value per channel when training, got input size torch.Size([1, 128]
因为用到了BatchNorm，所以batch_size要大于1，drop_last 参数设置为True。
如果是两个GPU训练，batch_size改为4。
参考
ValueError: Expected more than 1 value per channel when training, got input size [1, 16, 1, 1]（解决方案）
Pytorch遇到报错Expected more than 1 value per channel when training, got input size torch.Size
本文来自互联网用户投稿，文章观点仅代表作者本人，不代表本站立场，不承担相关法律责任。如若转载，请注明出处。 如若内容造成侵权/违法违规/事实不符，请点击【内容举报】进行投诉反馈！