码迷,mamicode.com
首页 > 其他好文 > 详细

pytorch中的 requires_grad和volatile

时间:2018-06-25 15:03:16      阅读:141      评论:0      收藏:0      [点我收藏+]

标签:UI   size   span   ORC   适用于   detail   font   更新   阶段   

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

 

简单总结其用途

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

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

 

pytorch中的 requires_grad和volatile

标签:UI   size   span   ORC   适用于   detail   font   更新   阶段   

原文地址:https://www.cnblogs.com/573177885qq/p/9223707.html

(0)
(0)
   
举报
评论 一句话评论(0
登录后才能评论!
© 2014 mamicode.com 版权所有  联系我们:gaon5@hotmail.com
迷上了代码!