PyTorch预训练的实现


Posted in Python onSeptember 18, 2019

前言

最近使用PyTorch感觉妙不可言,有种当初使用Keras的快感,而且速度还不慢。各种设计直接简洁,方便研究,比tensorflow的臃肿好多了。今天让我们来谈谈PyTorch的预训练,主要是自己写代码的经验以及论坛PyTorch Forums上的一些回答的总结整理。

直接加载预训练模型

如果我们使用的模型和原模型完全一样,那么我们可以直接加载别人训练好的模型:

my_resnet = MyResNet(*args, **kwargs)
my_resnet.load_state_dict(torch.load("my_resnet.pth"))

当然这样的加载方法是基于PyTorch推荐的存储模型的方法:

torch.save(my_resnet.state_dict(), "my_resnet.pth")

还有第二种加载方法:

my_resnet = torch.load("my_resnet.pth")

加载部分预训练模型

其实大多数时候我们需要根据我们的任务调节我们的模型,所以很难保证模型和公开的模型完全一样,但是预训练模型的参数确实有助于提高训练的准确率,为了结合二者的优点,就需要我们加载部分预训练模型。

pretrained_dict = model_zoo.load_url(model_urls['resnet152'])
model_dict = model.state_dict()
# 将pretrained_dict里不属于model_dict的键剔除掉
pretrained_dict = {k: v for k, v in pretrained_dict.items() if k in model_dict}
# 更新现有的model_dict
model_dict.update(pretrained_dict)
# 加载我们真正需要的state_dict
model.load_state_dict(model_dict)

因为需要剔除原模型中不匹配的键,也就是层的名字,所以我们的新模型改变了的层需要和原模型对应层的名字不一样,比如:resnet最后一层的名字是fc(PyTorch中),那么我们修改过的resnet的最后一层就不能取这个名字,可以叫fc_

微改基础模型预训练

对于改动比较大的模型,我们可能需要自己实现一下再加载别人的预训练参数。但是,对于一些基本模型PyTorch中已经有了,而且我只想进行一些小的改动那么怎么办呢?难道我又去实现一遍吗?当然不是。

我们首先看看怎么进行微改模型。

微改基础模型

PyTorch中的torchvision里已经有很多常用的模型了,可以直接调用:

  • AlexNet
  • VGG
  • ResNet
  • SqueezeNet
  • DenseNet
import torchvision.models as models
resnet18 = models.resnet18()
alexnet = models.alexnet()
squeezenet = models.squeezenet1_0()
densenet = models.densenet_161()

但是对于我们的任务而言有些层并不是直接能用,需要我们微微改一下,比如,resnet最后的全连接层是分1000类,而我们只有21类;又比如,resnet第一层卷积接收的通道是3, 我们可能输入图片的通道是4,那么可以通过以下方法修改:

resnet.conv1 = nn.Conv2d(4, 64, kernel_size=7, stride=2, padding=3, bias=False)
resnet.fc = nn.Linear(2048, 21)

简单预训练

模型已经改完了,接下来我们就进行简单预训练吧。
我们先从torchvision中调用基本模型,加载预训练模型,然后,重点来了,将其中的层直接替换为我们需要的层即可:

resnet = torchvision.models.resnet152(pretrained=True)
# 原本为1000类,改为10类
resnet.fc = torch.nn.Linear(2048, 10)

其中使用了pretrained参数,会直接加载预训练模型,内部实现和前文提到的加载预训练的方法一样。因为是先加载的预训练参数,相当于模型中已经有参数了,所以替换掉最后一层即可。OK!

以上就是本文的全部内容,希望对大家的学习有所帮助,也希望大家多多支持三水点靠木。

Python 相关文章推荐
进一步探究Python中的正则表达式
Apr 28 Python
分析用Python脚本关闭文件操作的机制
Jun 28 Python
python爬虫爬取网页表格数据
Mar 07 Python
Python File readlines() 使用方法
Mar 19 Python
Python操作Sql Server 2008数据库的方法详解
May 17 Python
transform python环境快速配置方法
Sep 27 Python
Python使用sorted对字典的key或value排序
Nov 15 Python
FFT快速傅里叶变换的python实现过程解析
Oct 21 Python
python Qt5实现窗体跟踪鼠标移动
Dec 13 Python
解决torch.autograd.backward中的参数问题
Jan 07 Python
python的pip有什么用
Jun 17 Python
python3通过subprocess模块调用脚本并和脚本交互的操作
Dec 05 Python
用python实现英文字母和相应序数转换的方法
Sep 18 #Python
Django模板导入母版继承和自定义返回Html片段过程解析
Sep 18 #Python
Python爬虫图片懒加载技术 selenium和PhantomJS解析
Sep 18 #Python
python rsa实现数据加密和解密、签名加密和验签功能
Sep 18 #Python
决策树剪枝算法的python实现方法详解
Sep 18 #Python
python生成requirements.txt的两种方法
Sep 18 #Python
python2与python3爬虫中get与post对比解析
Sep 18 #Python
You might like
PHP中is_file不能替代file_exists的理由
2014/03/04 PHP
ThinkPHP打开验证码页面显示乱码的解决方法
2014/12/18 PHP
php实现购物车功能(上)
2020/07/23 PHP
PHP实现的数独求解问题示例
2017/04/18 PHP
第一个JavaScript入门基础 document.write输出
2010/02/22 Javascript
前端开发必须知道的JS之原型和继承
2010/07/06 Javascript
javascript 模拟坦克大战游戏(html5版)附源码下载
2014/04/08 Javascript
jquery动态调整div大小使其宽度始终为浏览器宽度
2014/06/06 Javascript
jQuery实现鼠标经过时出现隐藏层文字链接的方法
2015/10/12 Javascript
实例详解jQuery的无new构建
2016/08/02 Javascript
bootstrap的工具提示实例代码
2017/05/17 Javascript
jQuery+ajax实现修改密码验证功能实例详解
2017/07/06 jQuery
使用D3.js制作图表详解
2017/08/13 Javascript
详解Web使用webpack构建前端项目
2017/09/23 Javascript
原生JavaScrpit中异步请求Ajax实现方法
2017/11/03 Javascript
jquery分页插件pagination使用教程
2018/10/23 jQuery
微信小程序实现元素渐入渐出动画效果封装方法
2019/05/18 Javascript
JS防抖和节流实例解析
2019/09/24 Javascript
原生JS实现弹幕效果的简单操作指南
2020/11/10 Javascript
全面解析Vue中的$nextTick
2020/12/24 Vue.js
Ruby使用eventmachine为HTTP服务器添加文件下载功能
2016/04/20 Python
解析Python中的__getitem__专有方法
2016/06/27 Python
Python数据结构与算法之列表(链表,linked list)简单实现
2017/10/30 Python
机器学习之KNN算法原理及Python实现方法详解
2018/07/09 Python
JupyterNotebook 输出窗口的显示效果调整实现
2020/09/22 Python
萌新的HTML5 入门指南
2020/11/06 HTML / CSS
美丽的珠宝配饰:SmallThings
2019/09/04 全球购物
会展中心部门工作职责
2013/11/27 职场文书
物流专业大学的自我评价
2014/01/11 职场文书
婚礼司仪主持词
2014/03/14 职场文书
中等生评语大全
2014/05/04 职场文书
班组建设经验交流材料
2014/05/12 职场文书
先进个人事迹材料(2016推荐版)
2016/03/01 职场文书
导游词之茶卡盐湖
2019/11/26 职场文书
手把手教你用SpringBoot将文件打包成zip存放或导出
2021/06/11 Java/Android
Python天气语音播报小助手
2021/09/25 Python