ValueError: Expected more than 1 value per channel when training, got input size [1, 256, 1, 1]的解决方案

在运行pytorch代码的时候，当其中存在BN层，并且你还存在输出的结果为1x1卷积，（如全局平均池化），在验证集的时候就很会出现这个问题：
ValueError: Expected more than 1 value per channel when training, got input size [1, 256, 1, 1]

可是这个时候检查一下batch_sizes 发现明明是大于1的，之所以还会出现这个，就是你数据的总数除以batch_size不能整除，所以需要我们在dataloader中将drop_last 参数设置为True。这样dataloader就会丢弃不能整除的部分，上面的问题也就得到解决