pytorch的raise ValueError(“Expected more than 1 value per channel when training, got input size {}“报错 – 小飞侠

pytorch的raise ValueError(“Expected more than 1 value per channel when training, got input size {}“报错

Post author:xfxia
Post published:2023年8月22日
Post category:其他

出现这个报错的原因是

模型中有BN模块，而模型处于训练模式，BN是拒绝任何batch-size为1的情况的

设置成model.eveal()即可

版权声明：本文为m0_56514535原创文章，遵循 CC 4.0 BY-SA 版权协议，转载请附上原文出处链接和本声明。

原文链接：https://blog.csdn.net/m0_56514535/article/details/131190950