博客指出在Pytorch训练中出现ValueError: Expected more than 1 value per channel when training, got input size [1, 256, 1, 1]错误，原因可能是输入批次只有一个数据点，BatchNorm操作需多于一个数据计算平均值。解决方法是在获取数据集时，将DataLoader中drop_last设置为True，丢弃不够一个批次的数据。
1.ValueError: Expected more than 1 value per channel when training, got input size [1, 512, 1, 1]
（参考：https://blog.youkuaiyun.com/sinat_39307513/article/details/87917537）
出现 ValueError: Expected more than 1 value per channel when training, got input size [1, 256, 1, 1] 问题，
可能是输入批次只有一个数据点，而由于BatchNorm操作需要多于一个数据计算平均值，因此造成该错误。
解决方法：
在获取数据集时，将DataLoader中drop_last设置为True。
把不够一个批次的数据丢弃。
扫一扫
5991
被折叠的 条评论
为什么被折叠?
到【灌水乐园】发言
