测试模型可用性时出现报错:
ValueError: Expected more than 1 value per channel when training, got input size torch.Size([1, 16, 1, 1])
看懂意思后明白是:由于batchnorm层需要大于一个样本去计算其中的参数,解决方法是将dataloader的一个丢弃参数设置为true。或者将batch的数量设置大于1即可。我这里将1改为10就可啦,只要是大于1的数就行。
ValueError: Expected more than 1 value per channel when training, got input size torch.Size([1, 16,
最新推荐文章于 2024-06-01 00:28:40 发布