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 相关文章推荐
在Django中创建URLconf相关的通用视图的方法
Jul 20 Python
浅析PHP与Python进行数据交互
May 15 Python
Python基于生成器迭代实现的八皇后问题示例
May 23 Python
PyQt弹出式对话框的常用方法及标准按钮类型
Feb 27 Python
Python基础之函数的定义与使用示例
Mar 23 Python
pymysql模块的使用(增删改查)详解
Sep 09 Python
Django自定义用户表+自定义admin后台中的字段实例
Nov 18 Python
Pandas时间序列重采样(resample)方法中closed、label的作用详解
Dec 10 Python
tensorflow 实现打印pb模型的所有节点
Jan 23 Python
在python中使用pyspark读写Hive数据操作
Jun 06 Python
python从ftp获取文件并下载到本地
Dec 05 Python
python解析json数据
Apr 29 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相关资料
2006/10/09 PHP
php xml常用函数的集合(比较详细)
2013/06/06 PHP
PHP框架Laravel插件Pagination实现自定义分页
2020/04/22 PHP
PHP合并数组的2种方法小结
2016/11/24 PHP
php写一个函数,实现扫描并打印出自定目录下(含子目录)所有jpg文件名
2017/05/26 PHP
PHP用函数嵌入网站访问量计数器
2017/10/27 PHP
PHP大文件分块上传功能实例详解
2019/07/22 PHP
PHP的垃圾回收机制代码实例讲解
2021/02/27 PHP
JavaScript 三种不同位置代码的写法
2009/10/25 Javascript
jquery限制输入字数,并提示剩余字数实现代码
2012/12/24 Javascript
window.open以post方式将内容提交到新窗口
2012/12/26 Javascript
uploadify在Firefox下丢失session问题的解决方法
2013/08/07 Javascript
jQuery修改class属性和CSS样式整理
2015/01/30 Javascript
Javascript实现网络监测的方法
2015/07/31 Javascript
Vue.js第四天学习笔记
2016/12/02 Javascript
BootStrap 图片样式、辅助类样式和CSS组件的实例详解
2017/01/20 Javascript
详解AngularJs HTTP响应拦截器实现登陆、权限校验
2017/04/11 Javascript
zTree树形菜单交互选项卡效果的实现方法
2017/12/25 Javascript
详解vue中async-await的使用误区
2018/12/05 Javascript
vue-cli3单页构建大型项目方案
2020/04/07 Javascript
javascript设计模式 ? 工厂模式原理与应用实例分析
2020/04/09 Javascript
vue2.x数组劫持原理的实现
2020/04/19 Javascript
es6函数之尾调用优化实例分析
2020/04/25 Javascript
介绍Python中的__future__模块
2015/04/27 Python
python实现将html表格转换成CSV文件的方法
2015/06/28 Python
如何将python中的List转化成dictionary
2016/08/15 Python
Python读取Excel表格,并同时画折线图和柱状图的方法
2018/10/14 Python
详解使用python爬取抖音app视频(appium可以操控手机)
2021/01/26 Python
CSS3 3D位移translate效果实例介绍
2016/05/03 HTML / CSS
HTML5 LocalStorage 本地存储刷新值还在
2017/03/10 HTML / CSS
T3官网:头发造型工具
2019/12/26 全球购物
2014年销售内勤工作总结
2014/12/01 职场文书
2015年团支部工作总结
2015/04/03 职场文书
活动主持人开场白
2015/05/28 职场文书
教师节班会开场白
2015/06/01 职场文书
青年岗位能手事迹材料(2016推荐版)
2016/03/01 职场文书