Pytorch 实现冻结指定卷积层的参数
作者:陈军2号 时间:2023-05-22 07:27:21
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
来源:https://blog.csdn.net/u011622208/article/details/86078055
标签:Pytorch,冻结,卷积层,参数
0
投稿
猜你喜欢
python中的列表与元组的使用
2023-07-23 08:25:12
xhtml+css页面制作过程中问题的解决方案
2008-08-05 18:00:00
PDO::beginTransaction讲解
2023-06-06 00:57:46
Python使用QQ邮箱发送邮件报错smtplib.SMTPAuthenticationError
2023-07-07 06:58:52
laravel入门知识点整理
2023-05-31 13:42:33
Asp 防止网页频繁刷新一法
2008-04-07 13:03:00
python wav模块获取采样率 采样点声道量化位数(实例代码)
2023-04-26 23:25:28
Python求正态分布曲线下面积实例
2021-01-28 18:20:07
python统计日志ip访问数的方法
2023-07-05 18:57:20
新版php study根目录下文件夹无法显示的图文解决方法
2023-11-15 00:44:29
支持汉转拼和拼音分词的PHP中文工具类ChineseUtil
2023-11-08 10:48:13
SQL存储过程介绍
2008-02-13 18:52:00
Microsoft SQL Server 2008 正式发布
2008-08-07 12:32:00
Opencv实现计算两条直线或线段角度方法详解
2023-10-01 22:18:15
Django实现CAS+OAuth2的方法示例
2023-01-26 17:32:46
在弹出窗口用POST提交数据
2010-02-24 09:39:00
INPUT的对齐问题
2008-05-24 08:32:00
另类解读SQL Server中的DateTime数据类型
2009-01-06 11:22:00
Python实现读取txt文件中的数据并绘制出图形操作示例
2021-07-21 17:01:15
Python中导入模块的几种方式总结
2023-08-04 17:03:54