在使用PyTorch训练神经网络时遇到ValueError，提示需要超过1个值每通道。问题可能源于两个方面：1) 批次大小(batch_size)设为1，BatchNorm需要多于一个样本计算均值；2) 数据集大小不能被batch_size整除，导致最后一批数据不足。解决方案是将batch_size设置为大于1的值，或者在DataLoader中设置drop_last为True来丢弃不足的批次。
在使用pytorch训练模型时，遇到下面的错误：
ValueError: Expected more than 1 value per channel when training, got input size [1, 16, 1, 1]
如图所示：
出现这个错误可能有两个原因：
1.输入批次只有一个数据点，而由于BatchNorm操作必须要多于一个数据去计算平均值，如把batch_size的值改为大于1的数
如图举例所示：
2.如果batch_size的值是大于1的还出现这个错误，可能是数据集的总数除以batch_size的值不能整除，即多余的数据不够一个批次。我们可以在dataloader中把drop_last 参数设置为True，这样dataloader就会舍去批次中多余的部分
如图举例所示：
扫一扫
被折叠的 条评论
为什么被折叠?
到【灌水乐园】发言
