Pytorch入坑

版权声明:本文为博主原创文章,遵循 CC 4.0 by-sa 版权协议,转载请附上原文出处链接和本声明。
本文链接:https://blog.csdn.net/supinyu/article/details/84666287

常见入坑

1、model.eval() 与torch.no_grad()的区别

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

展开阅读全文

没有更多推荐了,返回首页