Pytorch根据layers的name冻结训练方式


Posted in Python onJanuary 06, 2020

使用model.named_parameters()可以轻松搞定,

model.cuda()
 
 
# ######################################## Froze some layers to fine-turn the model ########################
for name, param in model.named_parameters(): # 带有参数名的模型的各个层包含的参数遍历
  if 'out' or 'merge' or 'before_regress' in name: # 判断参数名字符串中是否包含某些关键字
    continue
  param.requires_grad = False
# #############################################################################################################
 
 
optimizer = optim.SGD(filter(lambda p: p.requires_grad, model.parameters()),
           lr=opt.learning_rate * args.world_size, momentum=0.9, weight_decay=5e-4)

以上这篇Pytorch根据layers的name冻结训练方式就是小编分享给大家的全部内容了,希望能给大家一个参考,也希望大家多多支持三水点靠木。

Python 相关文章推荐
Python网络编程中urllib2模块的用法总结
Jul 12 Python
django+js+ajax实现刷新页面的方法
May 22 Python
python实现简单中文词频统计示例
Nov 08 Python
Python处理CSV与List的转换方法
Apr 19 Python
Python列表list排列组合操作示例
Dec 18 Python
Python3.5文件读与写操作经典实例详解
May 01 Python
python实现海螺图片的方法示例
May 12 Python
Python实现的文轩网爬虫完整示例
May 16 Python
Python 实现毫秒级淘宝抢购脚本的示例代码
Sep 16 Python
使用tqdm显示Python代码执行进度功能
Dec 08 Python
Python list运算操作代码实例解析
Jan 20 Python
python集合能干吗
Jul 19 Python
Windows下python3安装tkinter的问题及解决方法
Jan 06 #Python
带你彻底搞懂python操作mysql数据库(cursor游标讲解)
Jan 06 #Python
详解python tkinter模块安装过程
Jan 06 #Python
解决tensorflow由于未初始化变量而导致的错误问题
Jan 06 #Python
Python3 全自动更新已安装的模块实现
Jan 06 #Python
tensorflow 只恢复部分模型参数的实例
Jan 06 #Python
春节到了 教你使用python来抢票回家
Jan 06 #Python
You might like
一个简易需要注册的留言版程序
2006/10/09 PHP
坏狼的PHP学习教程之第2天
2008/06/15 PHP
PHP Class&Object -- 解析PHP实现二叉树
2013/06/25 PHP
使用PHP接收POST数据,解析json数据
2013/06/28 PHP
PHP判断文件是否被引入的方法get_included_files用法示例
2016/11/29 PHP
php 替换文章中的图片路径,下载图片到本地服务器的方法
2018/02/06 PHP
window.location和document.location的区别分析
2008/12/23 Javascript
修改jquery.lazyload.js实现页面延迟载入
2010/12/22 Javascript
一个基于jquery的文本框记数器
2012/09/19 Javascript
用jquery实现点击栏目背景色改变
2012/12/10 Javascript
jQuery动态修改超链接地址的方法
2015/02/13 Javascript
jquery选择器中的空格与大于号>、加号+与波浪号~的区别介绍
2016/06/24 Javascript
酷! 不同风格页面布局幻灯片特效js实现
2021/02/19 Javascript
webpack热模块替换(HMR)/热更新的方法
2018/04/05 Javascript
Vue父子组建的简单通信之控制开关Switch的实现
2018/06/04 Javascript
vue+swiper实现左右滑动的测试题功能
2020/10/30 Javascript
[01:01:04]2018DOTA2亚洲邀请赛 4.5 淘汰赛 OpTic vs TNC 第一场
2018/04/06 DOTA
[01:14]2019完美世界城市挑战赛(秋季赛)全国总决赛精彩花絮
2020/01/08 DOTA
pycharm中成功运行图片的配置教程
2018/10/28 Python
python监测当前联网状态并连接的实例
2018/12/18 Python
详解Python 中sys.stdin.readline()的用法
2019/09/12 Python
Python qrcode 生成一个二维码的实例详解
2020/02/12 Python
Windows下Anaconda和PyCharm的安装与使用详解
2020/04/23 Python
Python3 搭建Qt5 环境的方法示例
2020/07/16 Python
DHC中国官方购物网站:日本通信销售No.1化妆品
2016/08/20 全球购物
APM Monaco中国官网:来自摩纳哥珠宝品牌
2017/12/27 全球购物
bareMinerals官网:矿物质化妆品和护肤品
2018/02/04 全球购物
经济学人订阅:The Economist
2018/07/19 全球购物
《识字五》教学反思
2014/03/01 职场文书
银行内勤岗位职责
2014/04/09 职场文书
啤酒节策划方案
2014/05/28 职场文书
酒店爱岗敬业演讲稿
2014/09/02 职场文书
教师师德师风个人整改方案
2014/09/18 职场文书
2015年女生节活动总结
2015/02/27 职场文书
2015年公路养护工作总结
2015/05/13 职场文书
利用Python实现翻译HTML中的文本字符串
2022/06/21 Python