训练完resnet网络之后,想使用resnet直接对单张图片进行分类判别,但因为resnet存在BN层,导致出现报错“Expected more than 1 value per channel when training”,请问有什么方法解决?
可以参考这两篇文章,希望对你有帮助: https://blog.csdn.net/qq_42079689/article/details/102587401
https://www.cnblogs.com/zmbreathing/p/pyTorch_BN_error.html
model.eval(),将模型转成推理模式就可以了