解决ValueError: Expected more than 1 value per channel when training

出现这个问题是因为网络中存在BatchNormalization模块,它需要多于1个数据来计算平均值,当batch只有一个数据时会报错。
如果使用pytorch,可以在获取数据集时,将DataLoader中drop_last设置为True。把不够一个批次的数据丢弃。

原文 https://blog.csdn.net/sinat_39307513/article/details/87917537

你可能感兴趣的:(pytorch)