解决报错:ValueError: Expected more than 1 value per channel when training
在使用pytorch自定义一个图像分类网络的时候出现了上述的错误。查阅了下面的网址之后发现是batch normalization的函数出了问题。
我的解决方法和他们并不一样。batch normalization在batchsize为1的时候也会报这样的错误,将batchsize调大即可。
在使用pytorch自定义一个图像分类网络的时候出现了上述的错误。查阅了下面的网址之后发现是batch normalization的函数出了问题。
我的解决方法和他们并不一样。batch normalization在batchsize为1的时候也会报这样的错误,将batchsize调大即可。