在测试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的数就行。
在测试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的数就行。