- 博客(3)
- 收藏
- 关注
RSS订阅原创 PyCharm报错解决
a.既然知道BN在训练模式(training=True)时拒绝batch-size为1的情况，那么在训练的时候避免设置batch-size为1。在测试的时候把BN设置成测试模式(training=False)就好了，比如通过model.eval()方式。在nn.functional文件中，把刚刚的那两行判断batch-size为1就报错的代码注释掉。博主推荐两个方法，我用的第二种解决的。
2023-12-05 19:45:06
128
空空如也
空空如也
TA创建的收藏夹 TA关注的收藏夹
TA关注的人