AGI Hunt· cneuralnetwork·· 3 小时前AI 评分32
PyTorch 面试题:model.eval() 与 torch.no_grad() 的区别
PyTorch 面试题:model.eval() 与 torch.no_grad() 的区别
AI 导读
model.eval() 会改变 dropout、batch normalization 等层的行为,而 torch.no_grad() 用于关闭梯度追踪,两者分别作用于模型行为和 autograd 行为。普通推理场景通常需要同时使用这两个方法。
来源:AGI Hunt · agihunt.info