【学习笔记】关于volatile和0.4.0版本后的torch.set_grad_enabled()

volatile是用来在验证或者测试的时候将输入设置成volatile,这样后继几点全部都被设置成volatile,也就是不需要自动求导。

0.4.0以后采用torch.set_grad_enabled()来替代这种用法

但是torch.set_grad_enabled()在使用的时候是设置一个上下文环境,也就是说只要设置了torch.set_grad_enabled(False)那么接下来所有的tensor运算产生的新的节点都是不可求导的,这个相当于一个全局的环境,即使是多个循环或者是在函数内设置的调用,只要torch.set_grad_enabled(False)出现,则不管是在下一个循环里还是在主函数中,都不在求导,除非单独设置一个孤立节点,并把他的requires_grad设置成true。

评论 1
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

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

余额充值