文章目录 常见入坑1、model.eval() 与torch.no_grad()的区别 常见入坑 1、model.eval() 与torch.no_grad()的区别 model.eval() — 将通知你所有的图层你处于评估模式,这样,batchnorm或dropout图层将在eval模型而不是训练模式下工作 torch.no_grad() — 影响autograd引擎并停用它。它将减少内存使用并加快计算速度,但您将无法使用backprop(在eval脚本中不需要)