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编码时应该注意的几个情况
Mar 04 Python
在Python中处理字符串之isdecimal()方法的使用
May 20 Python
python比较两个列表大小的方法
Jul 11 Python
老生常谈Python startswith()函数与endswith函数
Sep 08 Python
python机器学习之神经网络(二)
Dec 20 Python
浅谈Python中eval的强大与危害
Mar 13 Python
python爬取百度贴吧前1000页内容(requests库面向对象思想实现)
Aug 10 Python
python读写csv文件的方法
Aug 13 Python
python多项式拟合之np.polyfit 和 np.polyld详解
Feb 18 Python
Python实现动态给类和对象添加属性和方法操作示例
Feb 29 Python
python判断正负数方式
Jun 03 Python
python批量生成身份证号到Excel的两种方法实例
Jan 14 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 return语句的另一个作用
2014/07/30 PHP
PHP中的事务使用实例
2015/05/26 PHP
php轻松实现文件上传功能
2016/03/03 PHP
php 如何获取文件的后缀名
2016/06/05 PHP
CI框架中类的自动加载问题分析
2016/11/21 PHP
PHP 计算两个特别大的整数实例代码
2018/05/07 PHP
php 处理png图片白色背景色改为透明色的实例代码
2018/12/10 PHP
Laravel的Auth验证Token验证使用自定义Redis的例子
2019/09/30 PHP
Wordpress ThickBox 点击图片显示下一张图的修改方法
2010/12/11 Javascript
jQuery 瀑布流 绝对定位布局(二)(延迟AJAX加载图片)
2012/05/23 Javascript
左侧是表头的JS表格控件(自写,网上没有的)
2013/06/04 Javascript
js利用prototype调用Array的slice方法示例
2014/06/09 Javascript
使用C++为node.js写扩展模块
2015/04/22 Javascript
可以浮动某个物体的jquery控件用法实例
2015/07/24 Javascript
js简单倒计时实现代码
2016/04/30 Javascript
详解Backbone.js框架中的模型Model与其集合collection
2016/05/05 Javascript
JS回调函数基本定义与用法实例分析
2017/05/24 Javascript
浅谈Express.js解析Post数据类型的正确姿势
2019/05/30 Javascript
JavaScript自定义超时API代码实例
2020/04/30 Javascript
微信小程序点击滚动到指定位置的实现
2020/05/22 Javascript
JS倒计时两种实现方式代码实例
2020/07/27 Javascript
js实现验证码干扰(动态)
2021/02/23 Javascript
python验证码识别的示例代码
2017/09/21 Python
python 获取utc时间转化为本地时间的方法
2018/12/31 Python
Python3之字节串bytes与字节数组bytearray的使用详解
2019/08/27 Python
python GUI库图形界面开发之PyQt5信号与槽多窗口数据传递详细使用方法与实例
2020/03/08 Python
CSS3实现水平居中、垂直居中、水平垂直居中的实例代码
2020/02/27 HTML / CSS
html5 datalist标签使用示例(自动完成组件)
2014/05/04 HTML / CSS
美国女孩服装购物网站:Justice
2017/03/04 全球购物
DJI美国:消费类无人机领域的领导者
2018/04/27 全球购物
学生检讨书怎么写
2015/05/07 职场文书
仰望星空观后感
2015/06/10 职场文书
小学英语新课改心得体会
2016/01/22 职场文书
redis配置文件中常用配置详解
2021/04/14 Redis
详解thinkphp的Auth类认证
2021/05/28 PHP
如何使用Tkinter进行窗口的管理与设置
2021/06/30 Python