使用pytorch中Resne网络的识别问题

Question

训练完resnet网络之后，想使用resnet直接对单张图片进行分类判别，但因为resnet存在BN层，导致出现报错“Expected more than 1 value per channel when training”，请问有什么方法解决？

Answer 1

可以参考这两篇文章，希望对你有帮助： https://blog.csdn.net/qq_42079689/article/details/102587401
https://www.cnblogs.com/zmbreathing/p/pyTorch_BN_error.html