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 相关文章推荐
Mac OS X10.9安装的Python2.7升级Python3.3步骤详解
Dec 04 Python
Python中获取网页状态码的两个方法
Nov 03 Python
初学python的操作难点总结(新手必看篇)
Aug 03 Python
Window10+Python3.5安装opencv的教程推荐
Apr 02 Python
python实时监控cpu小工具
Jun 21 Python
python使用wxpy轻松实现微信防撤回的方法
Feb 21 Python
Python 读取 YUV(NV12) 视频文件实例
Dec 09 Python
python plotly画柱状图代码实例
Dec 13 Python
python全局变量引用与修改过程解析
Jan 07 Python
Python调用C/C++的方法解析
Aug 05 Python
matplotlib之pyplot模块实现添加子图subplot的使用
Apr 25 Python
Python使用海龟绘图实现贪吃蛇游戏
Jun 18 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
ThinkPHP3.2.2的插件控制器功能简述
2014/07/09 PHP
JQuery 将元素显示在屏幕的中央的代码
2010/02/27 Javascript
NodeJs中的非阻塞方法介绍
2012/06/05 NodeJs
使用JavaScript动态设置样式实现代码及演示动画
2013/01/25 Javascript
js中生成map对象的方法
2014/01/09 Javascript
深入理解javascript构造函数和原型对象
2014/09/23 Javascript
JS中script标签defer和async属性的区别详解
2016/08/12 Javascript
基于JS实现checkbox全选功能实例代码
2016/10/31 Javascript
JavaScript之cookie技术详解
2016/11/18 Javascript
JS DOMReady事件的六种实现方法总结
2016/11/23 Javascript
jQuery插件DataTable使用方法详解(.Net平台)
2016/12/22 Javascript
JS实现数组去重复值的方法示例
2017/02/18 Javascript
WebSocket实现简单客服聊天系统
2017/05/12 Javascript
vue中的计算属性的使用和vue实例的方法示例
2017/12/04 Javascript
jquery写出PC端轮播图实例
2018/01/26 jQuery
详解js的作用域、预解析机制
2018/02/05 Javascript
小程序实现分类页
2019/07/12 Javascript
[46:25]DOTA2上海特级锦标赛主赛事日 - 4 败者组第五轮 MVP.Phx VS EG第二局
2016/03/05 DOTA
python模拟登录百度代码分享(获取百度贴吧等级)
2013/12/27 Python
Python EOL while scanning string literal问题解决方法
2020/09/18 Python
Python实现从URL地址提取文件名的方法
2015/05/15 Python
python Selenium爬取内容并存储至MySQL数据库的实现代码
2017/03/16 Python
Python结巴中文分词工具使用过程中遇到的问题及解决方法
2017/04/15 Python
python3.5 email实现发送邮件功能
2018/05/22 Python
Python匿名函数及应用示例
2019/04/09 Python
Pandas之Dropna滤除缺失数据的实现方法
2019/06/25 Python
python监控进程状态,记录重启时间及进程号的实例
2019/07/15 Python
对Django url的几种使用方式详解
2019/08/06 Python
python爬虫之爬取百度音乐的实现方法
2019/08/24 Python
css3弹性盒模型实例介绍
2013/05/27 HTML / CSS
波兰品牌内衣及泳装网上商店:Astratex.pl
2017/02/03 全球购物
实体的生命周期
2013/08/31 面试题
EntityManager都有哪些方法
2013/11/01 面试题
2015政治思想表现评语
2015/03/25 职场文书
python代码实现扫码关注公众号登录的实战
2021/11/01 Python
Alexa停服!网站排名将何去何从?目前还没有替代品。
2022/04/15 杂记