Pytorch修改ResNet模型全连接层进行直接训练实例


Posted in Python onSeptember 10, 2019

之前在用预训练的ResNet的模型进行迁移训练时,是固定除最后一层的前面层权重,然后把全连接层输出改为自己需要的数目,进行最后一层的训练,那么现在假如想要只是把

最后一层的输出改一下,不需要加载前面层的权重,方法如下:

model = torchvision.models.resnet18(pretrained=False)
num_fc_ftr = model.fc.in_features
model.fc = torch.nn.Linear(num_fc_ftr, 224)
model = nn.DataParallel(model, device_ids=config.gpus).to(device)

首先模型结构是必须要传入的,然后把最后一层的输出改为自己所需的数目

以上知识点很简单,大家可以测试下,感谢大家的阅读和对三水点靠木的支持。

Python 相关文章推荐
python append、extend与insert的区别
Oct 13 Python
Python 基础教程之包和类的用法
Feb 23 Python
详解Appium+Python之生成html测试报告
Jan 04 Python
python内存管理机制原理详解
Aug 12 Python
Python学习笔记之集合的概念和简单使用示例
Aug 22 Python
python批量将excel内容进行翻译写入功能
Oct 10 Python
Matplotlib scatter绘制散点图的方法实现
Jan 02 Python
Window版下在Jupyter中编写TensorFlow的环境搭建
Apr 10 Python
浅谈Python3多线程之间的执行顺序问题
May 02 Python
Python基于Tkinter编写crc校验工具
May 06 Python
Pytest实现setup和teardown的详细使用详解
Apr 17 Python
进行数据处理的6个 Python 代码块分享
Apr 06 Python
django drf框架自带的路由及最简化的视图
Sep 10 #Python
Pytorch中accuracy和loss的计算知识点总结
Sep 10 #Python
python3.7环境下安装Anaconda的教程图解
Sep 10 #Python
Windows10下 python3.7 安装 facenet的教程
Sep 10 #Python
python 图像处理画一个正弦函数代码实例
Sep 10 #Python
Python操作Mongodb数据库的方法小结
Sep 10 #Python
Python使用matplotlib绘制三维参数曲线操作示例
Sep 10 #Python
You might like
php 输出双引号"与单引号'的方法
2010/05/09 PHP
PHP输入流php://input介绍
2012/09/18 PHP
php中flush()、ob_flush()、ob_end_flush()的区别介绍
2013/02/17 PHP
PHP Opcache安装和配置方法介绍
2015/05/28 PHP
Thinkphp5 微信公众号token验证不成功的原因及解决方法
2017/11/12 PHP
PHP实现动态压缩js与css文件的方法
2018/05/02 PHP
yii2中关于加密解密的那些事儿
2018/06/12 PHP
laravel7学习之无限级分类的最新实现方法
2020/09/30 PHP
Jquery常用技巧收集整理篇
2010/11/14 Javascript
javascript开发技术大全-第1章javascript概述
2011/07/03 Javascript
JavaScript高级程序设计(第3版)学习笔记12 js正则表达式
2012/10/11 Javascript
JQuery伸缩导航练习示例
2013/11/13 Javascript
微信js-sdk界面操作接口用法示例
2016/10/12 Javascript
微信小程序实现图片自适应(支持多图)
2017/01/25 Javascript
让微信小程序支持ES6中Promise特性的方法详解
2017/06/13 Javascript
js计算最大公约数和最小公倍数代码实例
2019/09/11 Javascript
layui对工具条进行选择性的显示方法
2019/09/19 Javascript
使用uni-app开发微信小程序的实现
2019/12/13 Javascript
[04:41]2014DOTA2国际邀请赛 Liquid顺利突围晋级正赛
2014/07/09 DOTA
Python多线程原理与用法实例剖析
2019/01/22 Python
使用OpenCV实现仿射变换—缩放功能
2019/08/29 Python
Python @property使用方法解析
2019/09/17 Python
使用Nibabel库对nii格式图像的读写操作
2020/07/01 Python
python3 kubernetes api的使用示例
2021/01/12 Python
Timberland法国官网:购买靴子、鞋子、衣服、夹克和配饰
2019/11/30 全球购物
Farfetch中文官网:奢侈品牌时尚购物平台
2020/03/15 全球购物
简述进程的启动、终止的方式以及如何进行进程的查看
2013/07/12 面试题
实用求职信范文分享
2013/12/25 职场文书
生日宴会答谢词
2014/01/09 职场文书
演讲比赛获奖感言
2014/02/02 职场文书
优秀幼教自荐信
2014/02/03 职场文书
房地产广告策划方案
2014/05/15 职场文书
安全保卫工作竞聘材料
2014/08/25 职场文书
五心教育心得体会
2014/09/04 职场文书
党章培训心得体会
2014/09/04 职场文书
2014年光棍节活动策划方案(创意集锦)
2014/09/29 职场文书