训练完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
报告相同问题?