ValueError: Expected more than 1 value per channel when training, got input size torch.Size([1, 3200

在测试Parnet网络的时候报错

ValueError: Expected more than 1 value per channel when training, got input size torch.Size([1, 3200, 1, 1])

经过查阅资料得知：由于batchnorm层需要大于一个样本去计算其中的参数，解决方法是将dataloader的一个丢弃参数设置为true。或者将batch的数量设置大于1即可。我这里将1改为10就可啦，只要是大于1的数就行。