requires_grad: volatile 和 requires_grad在pytorch中的意思

45 篇文章 6 订阅
27 篇文章 5 订阅

https://blog.csdn.net/u012436149/article/details/66971822

在resnet.py文件中遇到了

Backward过程中排除子图

pytorchBP过程是由一个函数决定的,loss.backward(), 可以看到backward()函数里并没有传要求谁的梯度。
那么我们可以大胆猜测,在BP的过程中,pytorch是将所有影响lossVariable都求了一次梯度
但是有时候,我们并不想求所有Variable的梯度
那就要考虑如何在Backward过程中排除子图(ie.排除没必要的梯度计算)。 

如何BP过程中排除子图?
 Variable的两个参数(requires_gradvolatile

requires_grad=True   要求梯度
requires_grad=False   不要求梯度

volatile=True相当于requires_grad=False
反之则反之。。。。。。。ok

 

注意:
如果a是requires_grad=Trueb是requires_grad=False
c=a+b是requires_grad=True
同样的道理应用于volatile

 

为什么要排除子图

也许有人会问,梯度全部计算,不更新的话不就得了。 
这样就涉及了效率的问题了,计算很多没用的梯度是浪费了很多资源的(时间,计算机内存)

  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 打赏
    打赏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包

打赏作者

计算机视觉-Archer

图像分割没有团队的同学可加群

¥1 ¥2 ¥4 ¥6 ¥10 ¥20
扫码支付:¥1
获取中
扫码支付

您的余额不足,请更换扫码支付或充值

打赏作者

实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值