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单例模式实例分析
Jan 14 Python
Python类定义和类继承详解
May 08 Python
Python中线程编程之threading模块的使用详解
Jun 23 Python
python如何拆分含有多种分隔符的字符串
Mar 20 Python
使用python爬取B站千万级数据
Jun 08 Python
Python实现微信自动好友验证,自动回复,发送群聊链接方法
Feb 21 Python
Python实现最常见加密方式详解
Jul 13 Python
python config文件的读写操作示例
Sep 27 Python
Django实现celery定时任务过程解析
Apr 21 Python
解决安装新版PyQt5、PyQT5-tool后打不开并Designer.exe提示no Qt platform plugin的问题
Apr 24 Python
Python调用C语言程序方法解析
Jul 07 Python
Matplotlib 绘制饼图解决文字重叠的方法
Jul 24 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
谈一谈收音机的高放电路
2021/03/02 无线电
模仿OSO的论坛(三)
2006/10/09 PHP
编写漂亮的代码 - 将后台程序与前端程序分开
2008/04/23 PHP
PHP日期函数date格式化UNIX时间的方法
2015/03/19 PHP
PHP中加速、缓存扩展的区别和作用详解(eAccelerator、memcached、xcache、APC )
2016/07/09 PHP
iOS10推送通知开发教程
2016/09/19 PHP
用javascript实现分割提取页面所需内容
2007/05/09 Javascript
JavaScript定时器详解及实例
2013/08/01 Javascript
jQuery插件实现表格隔行换色且感应鼠标高亮行变色
2013/09/22 Javascript
javascript中style.left和offsetLeft的用法说明
2014/03/07 Javascript
利用函数的惰性载入提高javascript代码执行效率
2014/05/05 Javascript
jQuery防止重复绑定事件的解决方法
2016/05/14 Javascript
实例详解jQuery的无new构建
2016/08/02 Javascript
详解Weex基于Vue2.0开发模板搭建
2017/03/20 Javascript
ES6入门教程之Iterator与for...of循环详解
2017/05/17 Javascript
angularJS 发起$http.post和$http.get请求的实现方法
2017/05/18 Javascript
微信小程序之 catalog 切换实现解析
2019/09/12 Javascript
js实现时钟定时器
2020/03/26 Javascript
vue-cli3 引入 font-awesome的操作
2020/08/11 Javascript
[02:53]DOTA2亚洲邀请赛 NewBee战队巡礼
2015/02/03 DOTA
python中requests模块的使用方法
2015/04/08 Python
Python手机号码归属地查询代码
2016/05/04 Python
利用Python自带PIL库扩展图片大小给图片加文字描述的方法示例
2017/08/08 Python
Python实现的tcp端口检测操作示例
2018/07/24 Python
python爬虫简单的添加代理进行访问的实现代码
2019/04/04 Python
Python3.7 读取 mp3 音频文件生成波形图效果
2019/11/05 Python
Python中bisect的用法及示例详解
2020/07/20 Python
彪马美国官网:PUMA美国
2017/03/09 全球购物
LINUX下线程,GDI类的解释
2016/12/14 面试题
宝宝满月酒主持词和仪式流程
2014/03/27 职场文书
小学生暑假家长评语
2014/04/17 职场文书
基层组织建设年活动总结
2015/05/09 职场文书
毕业论文致谢词
2015/05/14 职场文书
大学组织委员竞选稿
2015/11/21 职场文书
python3实现Dijkstra算法最短路径的实现
2021/05/12 Python
浅谈如何提高PHP代码的质量
2021/05/28 PHP