首页 > 其他 > 详细

torch.no_grad

时间:2019-07-15 09:30:33      阅读:780      评论:0      收藏:0      [点我收藏+]

PyTorch里的requires_grad、volatile及no_grad

requires_grad=True   要求计算梯度

requires_grad=False   不要求计算梯度

with torch.no_grad()或者@torch.no_grad()中的数据不需要计算梯度,也不会进行反向传播

torch.no_grad()是新版本pytorch中volatile的替代)

 

torch.no_grad

原文:https://www.cnblogs.com/cekong/p/11177384.html

(0)
(0)
   
举报
评论 一句话评论(0
关于我们 - 联系我们 - 留言反馈 - 联系我们:wmxa8@hotmail.com
© 2014 bubuko.com 版权所有
打开技术之扣,分享程序人生!