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缩进和冒号详解
Jun 01 Python
Python入门_学会创建并调用函数的方法
May 16 Python
Python对列表去重的多种方法(四种方法)
Dec 05 Python
numpy使用fromstring创建矩阵的实例
Jun 15 Python
Python Dataframe 指定多列去重、求差集的方法
Jul 10 Python
python实现电子产品商店
Feb 26 Python
python 中如何获取列表的索引
Jul 02 Python
django 配置阿里云OSS存储media文件的例子
Aug 20 Python
python在OpenCV里实现投影变换效果
Aug 30 Python
Python执行时间的几种计算方法
Jul 31 Python
Django路由层URLconf作用及原理解析
Sep 24 Python
python 获取剪切板内容的两种方法
Nov 28 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 header功能的使用
2013/10/28 PHP
PHP邮件发送类PHPMailer用法实例详解
2014/09/22 PHP
php使用memcoder将视频转成mp4格式的方法
2015/03/12 PHP
PHP中异常处理的一些方法整理
2015/07/03 PHP
php实现批量删除挂马文件及批量替换页面内容完整实例
2016/07/08 PHP
zen cart实现订单中增加paypal中预留电话的方法
2016/07/12 PHP
关于 Laravel Redis 多个进程同时取队列问题详解
2017/12/25 PHP
PHP常用字符串函数小结(推荐)
2018/08/05 PHP
laravel配置Redis多个库的实现方法
2019/04/10 PHP
转一个日期输入控件,支持FF
2007/04/27 Javascript
Ext面向对象开发实践(续)
2008/11/18 Javascript
Dojo 学习要点
2010/09/03 Javascript
根据IP的地址,区分不同的地区,查看不同的网站页面的js代码
2013/02/26 Javascript
JQuery解析HTML、JSON和XML实例详解
2014/03/29 Javascript
nodejs npm包管理的配置方法及常用命令介绍
2014/06/05 NodeJs
使用jQuery mobile库检测url绝对地址和相对地址的方法
2015/12/04 Javascript
js简单判断移动端系统的方法
2016/02/25 Javascript
Bootstrap布局组件教程之Bootstrap下拉菜单
2016/06/12 Javascript
JavaScript取得gridview中获取checkbox选中的值
2017/07/24 Javascript
详解关于Vue2.0路由开启keep-alive时需要注意的地方
2018/09/18 Javascript
微信小程序实现左滑修改、删除功能
2020/10/19 Javascript
vue项目上传Github预览的实现示例
2018/11/06 Javascript
小程序实现图片移动缩放效果
2020/05/26 Javascript
Openlayers实现地图的基本操作
2020/09/28 Javascript
基于vue与element实现创建试卷相关功能(实例代码)
2020/12/07 Vue.js
python标准日志模块logging的使用方法
2013/11/01 Python
用python结合jieba和wordcloud实现词云效果
2017/09/05 Python
使用python获取邮箱邮件的设置方法
2019/09/20 Python
Numpy将二维数组添加到空数组的实现
2019/12/05 Python
python类中super() 的使用解析
2019/12/19 Python
用 Python 制作地球仪的方法
2020/04/24 Python
DJI大疆德国官方商城:大疆无人机
2018/09/01 全球购物
说出一些常用的类,包,接口
2014/09/22 面试题
运动会跳远加油稿
2014/02/20 职场文书
2014大学生批评与自我批评思想汇报
2014/09/21 职场文书
2015年班级工作总结范文
2015/04/03 职场文书