ValueError: Expected more than 1 value per channel when training, got input size torch.Size([1, 72, 1, 1])
在跑模型的时候突然出现这个问题,我有很多数据集,模型到其中一个数据集的时候都没问题,就跑这个会有问题,以前没遇到过,今天尝试解决了。
主要出现在BN层中
修改数据集数量,让数据集数量除以batchsize余数为0或者大于1。后来问同门,在dataloder中可以设置如果除不尽直接舍弃余数。