pytorch中的 requires_grad和volatile

Posted 573177885qq

tags:

篇首语:本文由小常识网(cha138.com)小编为大家整理,主要介绍了pytorch中的 requires_grad和volatile相关的知识,希望对你有一定的参考价值。

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

 

简单总结其用途

(1)requires_grad=Fasle时不需要更新梯度, 适用于冻结某些层的梯度;

 (2)volatile=True相当于requires_grad=False,适用于推断阶段,不需要反向传播

 

以上是关于pytorch中的 requires_grad和volatile的主要内容,如果未能解决你的问题,请参考以下文章

了解 PyTorch 中的反向传播

PyTorch中的backward [转]

pytorch 如何设置 .requires_grad False

Pytorch实现线性回归

Pytorch实现线性回归(手动完成)

pytorch学习2