PyTorch加载预训练模型实例(pretrained)


Posted in Python onJanuary 17, 2020

使用预训练模型的代码如下:

# 加载预训练模型
 resNet50 = models.resnet50(pretrained=True)
 ResNet50 = ResNet(Bottleneck, [3, 4, 6, 3], num_classes=2)

 # 读取参数
 pretrained_dict = resNet50.state_dict()
 model_dict = ResNet50.state_dict()

 # 将pretained_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
 ResNet50.load_state_dict(model_dict)

以上这篇PyTorch加载预训练模型实例(pretrained)就是小编分享给大家的全部内容了,希望能给大家一个参考,也希望大家多多支持三水点靠木。

Python 相关文章推荐
Python中的defaultdict模块和namedtuple模块的简单入门指南
Apr 01 Python
python 安装virtualenv和virtualenvwrapper的方法
Jan 13 Python
python如何使用unittest测试接口
Apr 04 Python
Python3.6连接Oracle数据库的方法详解
May 18 Python
解决pandas 作图无法显示中文的问题
May 24 Python
使用python将图片按标签分入不同文件夹的方法
Dec 08 Python
python使用插值法画出平滑曲线
Dec 15 Python
浅谈django url请求与数据库连接池的共享问题
Aug 29 Python
Python opencv相机标定实现原理及步骤详解
Apr 09 Python
Django通过json格式收集主机信息
May 29 Python
python zip()函数的使用示例
Sep 23 Python
python实现文件分片上传的接口自动化
Nov 19 Python
python 正则表达式参数替换实例详解
Jan 17 #Python
Python函数式编程实例详解
Jan 17 #Python
python实现tail -f 功能
Jan 17 #Python
解决Python命令行下退格,删除,方向键乱码(亲测有效)
Jan 16 #Python
python对象销毁实例(垃圾回收)
Jan 16 #Python
python读取dicom图像示例(SimpleITK和dicom包实现)
Jan 16 #Python
.dcm格式文件软件读取及python处理详解
Jan 16 #Python
You might like
c#中的实现php中的preg_replace
2009/12/21 PHP
php中常量DIRECTORY_SEPARATOR用法深入分析
2014/11/14 PHP
CI框架常用方法小结
2016/05/17 PHP
Laravel5.* 打印出执行的sql语句的方法
2017/07/24 PHP
Laravel 微信小程序后端搭建步骤详解
2019/11/26 PHP
浅谈JavaScript中面向对象技术的模拟
2006/09/25 Javascript
Javascript 判断函数类型完美解决方案
2009/09/02 Javascript
javascript 匿名函数的理解(透彻版)
2010/01/28 Javascript
前台js对象在后台转化java对象的问题探讨
2013/12/20 Javascript
JS实现鼠标点击展开或隐藏表格行的方法
2015/03/03 Javascript
配置Grunt的Task时通配符支持和动态生成文件名问题
2015/09/06 Javascript
图片旋转、鼠标滚轮缩放、镜像、切换图片js代码
2020/12/13 Javascript
js获取iframe中的window对象的实现方法
2016/05/20 Javascript
微信小程序 教程之WXML
2016/10/18 Javascript
使用jquery实现的循环连续可停顿滚动实例
2016/11/23 Javascript
easyui combogrid实现本地模糊搜索过滤多列
2017/05/13 Javascript
利用ECharts.js画K线图的方法示例
2018/01/10 Javascript
JSON数据中存在单个转义字符“\”的处理方法
2018/07/11 Javascript
jQuery实现参数自定义的文字跑马灯效果
2018/08/15 jQuery
详解vue中localStorage的使用方法
2018/11/22 Javascript
js简单实现自动生成表格功能示例
2020/06/02 Javascript
[01:20:37]FNATIC vs NIP 2019国际邀请赛小组赛 BO2 第一场 8.16
2019/08/19 DOTA
python3爬虫之设计签名小程序
2018/06/19 Python
pymysql 开启调试模式的实现
2019/09/24 Python
matplotlib 对坐标的控制,加图例注释的操作
2020/04/17 Python
Python如何定义接口和抽象类
2020/07/28 Python
Python延迟绑定问题原理及解决方案
2020/08/04 Python
澳大利亚100%丝绸多彩度假装商店:TheSwankStore
2019/09/04 全球购物
LN-CC中国:高端男装和女装的奢侈时尚目的地
2019/09/14 全球购物
英语专业学生个人求职信范文
2014/01/06 职场文书
西门豹教学反思
2014/02/04 职场文书
意外伤害赔偿协议书范本
2014/09/28 职场文书
2014工程部年度工作总结
2014/12/17 职场文书
鼋头渚导游词
2015/02/05 职场文书
英语读书笔记
2015/07/02 职场文书
python 爬取华为应用市场评论
2021/05/29 Python