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计算方程式根的方法
May 07 Python
python 排序算法总结及实例详解
Sep 28 Python
Python面向对象之继承代码详解
Jan 29 Python
使用python生成目录树
Mar 29 Python
对python 矩阵转置transpose的实例讲解
Apr 17 Python
Pandas:DataFrame对象的基础操作方法
Jun 07 Python
Python + OpenCV 实现LBP特征提取的示例代码
Jul 11 Python
python使用原始套接字发送二层包(链路层帧)的方法
Jul 22 Python
python实现微信打飞机游戏
Mar 24 Python
Python爬虫实例——爬取美团美食数据
Jul 15 Python
Prometheus开发中间件Exporter过程详解
Nov 30 Python
Python实战之实现简易的学生选课系统
May 25 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
第五章 php数组操作
2011/12/30 PHP
php读取文件内容的几种方法详解
2013/06/26 PHP
PHP Yii框架之表单验证规则大全
2015/11/16 PHP
传智播客学习之JavaScript基础篇
2009/11/13 Javascript
jQuery源码分析之Event事件分析
2010/06/07 Javascript
元素的内联事件处理函数的特殊作用域在各浏览器中存在差异
2011/01/12 Javascript
JQuery弹出炫丽对话框的同时让背景变灰色
2014/05/22 Javascript
完美兼容各大浏览器的jQuery仿新浪图文淡入淡出间歇滚动特效
2014/11/12 Javascript
jQuery实现友好的轮播图片特效
2015/01/12 Javascript
安装使用Mongoose配合Node.js操作MongoDB的基础教程
2016/03/01 Javascript
jQuery事件与动画基础详解
2017/02/23 Javascript
超级简易的JS计算器实例讲解(实现加减乘除)
2017/08/08 Javascript
ES6中Array.includes()函数的用法
2017/09/20 Javascript
vue2.0 获取从http接口中获取数据,组件开发,路由配置方式
2019/11/04 Javascript
python使用PyFetion来发送短信的例子
2014/04/22 Python
python实现删除文件与目录的方法
2014/11/10 Python
用Python中的字典来处理索引统计的方法
2015/05/05 Python
Python制作爬虫采集小说
2015/10/25 Python
Python的MongoDB模块PyMongo操作方法集锦
2016/01/05 Python
Python实战小程序利用matplotlib模块画图代码分享
2017/12/09 Python
Python中将变量按行写入txt文本中的方法
2018/04/03 Python
Python实现模拟登录网易邮箱的方法示例
2018/07/05 Python
Python Gluon参数和模块命名操作教程
2019/12/18 Python
在python中使用nohup命令说明
2020/04/16 Python
HTML5不支持标签和新增标签详解
2016/06/27 HTML / CSS
打印机墨盒:123Inkjets
2017/02/16 全球购物
GUESS西班牙官方网上商城:美国服饰品牌
2017/03/15 全球购物
财务专业大学生职业生涯规划范文
2013/12/30 职场文书
人事助理自荐信
2014/02/02 职场文书
40岁生日感言
2014/02/15 职场文书
幼儿园毕业寄语
2014/04/03 职场文书
工业设计毕业生自荐信
2014/04/13 职场文书
新婚姻法离婚协议书范文
2014/11/30 职场文书
优秀校长事迹材料
2014/12/24 职场文书
文案策划岗位职责
2015/02/11 职场文书
python接口测试返回数据为字典取值方式
2022/02/12 Python