pytorch 预训练层的使用方法


Posted in Python onAugust 20, 2019

pytorch 预训练层的使用方法

将其他地方训练好的网络,用到新的网络里面

加载预训练网络

1.原先已经训练好一个网络 AutoEncoder_FC()

2.首先加载该网络,读取其存储的参数

3.设置一个参数集

cnnpre = AutoEncoder_FC()
cnnpre.load_state_dict(torch.load('autoencoder_FC.pkl')['state_dict'])
cnnpre_dict =cnnpre.state_dict()

加载新网络

1.设置新的网络

2.设置新网络参数集

cnn= AutoEncoder()
cnn_dict = cnn.state_dict()

更新新网络参数

1.将两个参数集比对,存在的网络参数保留

2.使用保留下的参数更新新网络参数集

3.加载新网络参数集到新网络中

cnnpre_dict = {k: v for k, v in cnnpre_dict.items() if k in cnn_dict}
cnn_dict.update(cnnpre_dict)
cnn.load_state_dict(cnn_dict)

以上这篇pytorch 预训练层的使用方法就是小编分享给大家的全部内容了,希望能给大家一个参考,也希望大家多多支持三水点靠木。

Python 相关文章推荐
python中将阿拉伯数字转换成中文的实现代码
May 19 Python
Python的Django中django-userena组件的简单使用教程
May 30 Python
Python中文字符串截取问题
Jun 15 Python
详解python之简单主机批量管理工具
Jan 27 Python
用python实现的线程池实例代码
Jan 06 Python
python使用TensorFlow进行图像处理的方法
Feb 28 Python
Python图像的增强处理操作示例【基于ImageEnhance类】
Jan 03 Python
python实现定时压缩指定文件夹发送邮件
Dec 22 Python
Python 多线程搜索txt文件的内容,并写入搜到的内容(Lock)方法
Aug 23 Python
简单介绍django提供的加密算法
Dec 18 Python
Python用户自定义异常的实现
Dec 25 Python
python中yield的用法详解
Jan 13 Python
python爬虫 urllib模块反爬虫机制UA详解
Aug 20 #Python
Pytorch 抽取vgg各层并进行定制化处理的方法
Aug 20 #Python
python实现抠图给证件照换背景源码
Aug 20 #Python
python爬虫 基于requests模块发起ajax的get请求实现解析
Aug 20 #Python
pytorch 在sequential中使用view来reshape的例子
Aug 20 #Python
pytorch在fintune时将sequential中的层输出方法,以vgg为例
Aug 20 #Python
python实现证件照换底功能
Aug 20 #Python
You might like
第二章 PHP入门基础之php代码写法
2011/12/30 PHP
Mootools 1.2教程 同时进行多个形变动画
2009/09/15 Javascript
Javascript的闭包
2009/12/31 Javascript
几个比较实用的JavaScript 测试及效验工具
2010/04/18 Javascript
用js实现判断当前网址的来路如果不是指定的来路就跳转到指定页面
2011/05/02 Javascript
JS的事件绑定深入认识
2014/06/26 Javascript
JQuery悬停控制图片轮播——代码简单
2015/08/05 Javascript
Javascript技术难点之apply,call与this之间的衔接
2015/12/04 Javascript
js禁止表单重复提交
2017/08/29 Javascript
解决Vue编译时写在style中的路径问题
2017/09/21 Javascript
微信小程序前端promise封装代码实例
2019/08/24 Javascript
vue项目中引入Sass实例方法
2019/08/27 Javascript
python通过wxPython打开一个音频文件并播放的方法
2015/03/25 Python
Python的GUI框架PySide的安装配置教程
2016/02/16 Python
玩转python爬虫之正则表达式
2016/02/17 Python
Python实现简单生成验证码功能【基于random模块】
2018/02/10 Python
Python中typing模块与类型注解的使用方法
2019/08/05 Python
python 数据提取及拆分的实现代码
2019/08/26 Python
Python3 A*寻路算法实现方式
2019/12/24 Python
Pandas缺失值2种处理方式代码实例
2020/06/13 Python
英国领先的独立时装店:Van Mildert
2019/10/28 全球购物
光声世纪笔试题目
2012/08/25 面试题
程序员经常用到的UNIX命令
2015/04/13 面试题
一道Delphi面试题
2016/10/28 面试题
市场营销专业个人自荐信格式
2013/09/21 职场文书
厨房工作人员岗位职责
2013/11/15 职场文书
影视制作岗位职责
2013/12/04 职场文书
应聘护理专业毕业自荐书范文
2014/02/12 职场文书
双创工作实施方案
2014/03/26 职场文书
文明寄语大全
2014/04/11 职场文书
老干部工作汇报材料
2014/10/28 职场文书
小学优秀教师先进事迹材料
2014/12/16 职场文书
绍兴鲁迅故居导游词
2015/02/09 职场文书
工作自我评价范文
2015/03/05 职场文书
婚宴新郎致辞
2015/07/28 职场文书
Java日常练习题,每天进步一点点(38)
2021/07/26 Java/Android