跳到正文
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