今天调代码,发现测试的时候同一个输入向量,前后测试两次网络的输出结果居然不一样,后来发现是网络中用了BN层,在测试的时候,如果不用model.eval()函数固定网络参数,那么测试的时候还是会计算均值和方差,就会导致输出的结果每一次都在变化。而且当输入一个样本进行测试的时候,还会报如上的错误。
Pytorch使用batchNormalization的时候遇到报错:ValueError: Expected more than 1 value per channel when training
最新推荐文章于 2024-01-23 13:09:41 发布