torch.set_grad_enabled(False)

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。
 

你可能感兴趣的:(pytorch,python)