因为用到了BatchNorm，所以batch_size要大于1，drop_last 参数设置为True。
如果是两个GPU训练，batch_size改为4。文章来源：https://www.toymoban.com/news/detail-792740.html
参考
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文章来源地址https://www.toymoban.com/news/detail-792740.html
到了这里，关于ValueError: Expected more than 1 value per channel when training, got input size torch.Size([1, 128]的文章就介绍完了。如果您还想了解更多内容，请在右上角搜索TOY模板网以前的文章或继续浏览下面的相关文章，希望大家以后多多支持TOY模板网！
