今天小编就为大家分享一篇Pytorch 实现冻结指定卷积层的参数,具有很好的参考价值,希望对大家有所帮助。一起跟随小编过来看看吧
python代码
for i, para in enumerate(self._net.module.features.parameters()): if i
分析
通过for循环将需要冻结的layer的requires_grad属性设置为False
以上这篇Pytorch 实现冻结指定卷积层的参数就是小编分享给大家的全部内容了,希望能给大家一个参考,也希望大家多多支持gaodaima搞代码网。
以上就是Pytorch 实现冻结指定卷积层的参数的详细内容,更多请关注gaodaima
来源gao!daima.com搞$代!码网
搞代码网其它相关文章!