- 博客(2)
- 收藏
- 关注
RSS订阅翻译 torch.no_grad()和nn.Module.eval()区别
当你需要执行不应该被autograd记录的操作，但你仍然希望以后在grad模式下使用这些计算的输出时，启用no-grad模式。model.eval()如何影响你的模型，完全取决于你的模型中使用的具体模块，以及它们是否定义了任何训练模式下的特定行为。建议你在训练时总是使用model.train()，在评估你的模型（验证/测试）时总是使用model.eval()，即使你不确定你的模型有训练模式的特定行为，因为你正在使用的模块可能被更新为在训练和评估模式下的不同行为。
2023-02-18 15:07:33
964
空空如也
空空如也
TA创建的收藏夹 TA关注的收藏夹
TA关注的人