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使用BeautifulSoup分析网页信息的方法
Apr 04 Python
python类继承用法实例分析
May 27 Python
分析用Python脚本关闭文件操作的机制
Jun 28 Python
PyQt5 QSerialPort子线程操作的实现
Apr 21 Python
python保存二维数组到txt文件中的方法
Nov 15 Python
Django logging配置及使用详解
Jul 23 Python
pygame实现贪吃蛇游戏(下)
Oct 29 Python
flask 实现token机制的示例代码
Nov 07 Python
Python3 使用map()批量的转换数据类型,如str转float的实现
Nov 29 Python
如何在Python 游戏中模拟引力
Mar 27 Python
Python基础之pandas数据合并
Apr 27 Python
python中pandas对多列进行分组统计的实现
Jun 18 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
探讨:web上存漏洞及原理分析、防范方法
2013/06/29 PHP
Smarty模板变量调节器用法分析
2016/05/23 PHP
浅谈Yii乐观锁的使用及原理
2017/07/25 PHP
用JS实现一个页面多个css样式实现
2008/05/29 Javascript
javascript 跳转代码集合
2009/12/03 Javascript
js 异步操作回调函数如何控制执行顺序
2013/12/24 Javascript
js实现拖拽效果
2015/02/12 Javascript
JS中捕获console.log()输出的方法
2015/04/16 Javascript
JS实现统计复选框选中个数并提示确定与取消的方法
2015/07/01 Javascript
jQuery form 表单验证插件(fieldValue)校验表单
2016/01/24 Javascript
详解Javascript ES6中的箭头函数(Arrow Functions)
2016/08/24 Javascript
es6的数字处理的方法(5个)
2017/03/16 Javascript
javascript 中select框触发事件过程的分析
2017/08/01 Javascript
Vue.js 实现微信公众号菜单编辑器功能(一)
2018/05/08 Javascript
angular 组件通信的几种实现方式
2018/07/13 Javascript
angularJs中$scope数据序列化的实例
2018/09/30 Javascript
微信小程序事件对象中e.target和e.currentTarget的区别详解
2019/05/08 Javascript
详解使用JWT实现单点登录(完全跨域方案)
2019/08/02 Javascript
javascript中innerHTML 获取或替换html内容的实现代码
2020/03/17 Javascript
微信小程序canvas动态时钟
2020/10/22 Javascript
通过实例解析js可枚举属性与不可枚举属性
2020/12/02 Javascript
python实现在windows下操作word的方法
2015/04/28 Python
Python 实现两个列表里元素对应相乘的方法
2018/11/14 Python
Python实现九宫格式的朋友圈功能内附“马云”朋友圈
2019/05/07 Python
详解python中的time和datetime的常用方法
2019/07/08 Python
python Tcp协议发送和接收信息的例子
2019/07/22 Python
在服务器上安装python3.8.2环境的教程详解
2020/04/26 Python
解决Alexnet训练模型在每个epoch中准确率和loss都会一升一降问题
2020/06/17 Python
Pytorch框架实现mnist手写库识别(与tensorflow对比)
2020/07/20 Python
css3弹性盒模型实例介绍
2013/05/27 HTML / CSS
Europcar美国/加拿大:预订汽车或卡车租赁服务
2018/11/13 全球购物
验房委托书
2014/08/30 职场文书
2014年小学体育工作总结
2014/12/11 职场文书
副总经理岗位职责范本
2015/04/08 职场文书
2015年党员发展工作总结
2015/05/13 职场文书
高中同学会致辞
2015/08/01 职场文书