服务器之家

服务器之家 > 正文

Pytorch 实现冻结指定卷积层的参数

时间:2020-05-09 09:43     来源/作者:陈军2号

python代码

?
1
2
3
4
5
6
7
8
9
10
11
12
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 实现冻结指定卷积层的参数就是小编分享给大家的全部内容了,希望能给大家一个参考,也希望大家多多支持服务器之家。

原文链接:https://blog.csdn.net/u011622208/article/details/86078055

相关文章

热门资讯

2022年最旺的微信头像大全 微信头像2022年最新版图片
2022年最旺的微信头像大全 微信头像2022年最新版图片 2022-01-10
蜘蛛侠3英雄无归3正片免费播放 蜘蛛侠3在线观看免费高清完整
蜘蛛侠3英雄无归3正片免费播放 蜘蛛侠3在线观看免费高清完整 2021-08-24
背刺什么意思 网络词语背刺是什么梗
背刺什么意思 网络词语背刺是什么梗 2020-05-22
yue是什么意思 网络流行语yue了是什么梗
yue是什么意思 网络流行语yue了是什么梗 2020-10-11
暖暖日本高清免费中文 暖暖在线观看免费完整版韩国
暖暖日本高清免费中文 暖暖在线观看免费完整版韩国 2021-05-08
返回顶部