Pytorch 实现冻结指定卷积层的参数
更新时间：2020年01月06日 09:36:19 作者：陈军2号
今天小编就为大家分享一篇Pytorch 实现冻结指定卷积层的参数，具有很好的参考价值，希望对大家有所帮助。一起跟随小编过来看看吧
python代码
for i, para in enumerate(self._net.module.features.parameters()):
if i < 16:
para.requires_grad = False
else:
para.requires_grad = True
# Solver.
# self._solver = torch.optim.SGD(
# self._net.parameters(), lr=self._options['base_lr'],
# momentum=0.9, weight_decay=self._options['weight_decay'])
self._solver = torch.optim.SGD(
self._net.module.parameters(), lr=self._options['base_lr'],
momentum=0.9, weight_decay=self._options['weight_decay'])
分析
通过for循环将需要冻结的layer的requires_grad属性设置为False
以上这篇Pytorch 实现冻结指定卷积层的参数就是小编分享给大家的全部内容了，希望能给大家一个参考，也希望大家多多支持脚本之家。
相关文章
这篇文章主要为大家详细介绍了Python机器学习中实现距离和相似性计算的方法，文中的示例代码讲解详细，感兴趣的小伙伴可以跟随小编一起学习一下2023-03-03
这篇文章主要介绍了Python中关于set的基本用法，具有很好的参考价值，希望对大家有所帮助。如有错误或未考虑完全的地方，望不吝赐教2023-04-04
这篇文章主要介绍了Python预测2020高考分数和录取情况可能是这样,本文通过示例代码给大家介绍的非常详细，对大家的学习或工作具有一定的参考借鉴价值，需要的朋友可以参考下2020-07-07
在python中如何实现将带小数的浮点型字符串转换为整数呢？今天小编就为大家介绍一下解决方案，希望对大家有所帮助。一起跟随小编过来看看吧2021-05-05
这篇文章主要介绍了基于python实现cdn日志文件导入mysql进行分析，本文以阿里云CDN日志作为辅助查询数据展开主题内容，其它云平台大同小异，需要的小伙伴可以参考一下2022-05-05
这篇文章主要为大家详细介绍了Flask Web开发入门之文件上传的相关资料，具有一定的参考价值，感兴趣的小伙伴们可以参考一下2018-08-08
这篇文章主要介绍了基于Numba提高python运行效率过程解析,文中通过示例代码介绍的非常详细，对大家的学习或者工作具有一定的参考学习价值,需要的朋友可以参考下2020-03-03
心情不好的时候，来用Python和Turtle库画棵樱花树送给自己吧，自己也要好好爱自己才对！文中的示例代码讲解详细，感兴趣的小伙伴可以动手试一试2022-02-02
这篇文章主要为大家详细介绍了python使用多线程编写tcp客户端程序，具有一定的参考价值，感兴趣的小伙伴们可以参考一下2019-09-09
range()是python内置函数它能返回一系列连续增加的整数，它的工作方式类似于分片，可以生成一个列表对象,这篇文章主要介绍了Python range() 函数用法,需要的朋友可以参考下2023-03-03
最新评论