- 博客(1)
- 收藏
- 关注
原创 model.train()&&model.eval()&&with torch.no_grad()用法
启用 Batch Normalization 和 Dropout如果模型中有BN层(Batch Normalization)和Dropout,需要在训练时添加model.train()。model.train()是保证BN层能够用到每一批数据的均值和方差。对于Dropout,model.train()是随机取一部分网络连接来训练更新参数。用法:注意model.train()的位置model.train()一定要是在epoch中并且在dataloader循环前的位置上2.model.eval不
2022-06-12 21:16:11
1010
3
空空如也
空空如也
TA创建的收藏夹 TA关注的收藏夹
TA关注的人