Pytorch修改ResNet模型全连接层进行直接训练实例
作者:嶙羽 时间:2023-06-07 03:49:28
之前在用预训练的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)
首先模型结构是必须要传入的,然后把最后一层的输出改为自己所需的数目
以上知识点很简单,大家可以测试下,感谢大家的阅读和对脚本之家的支持。
来源:https://www.cnblogs.com/yqpy/p/11497290.html
标签:Pytorch,ResNet模型
0
投稿
猜你喜欢
成功的用户界面的八个特性[译]
2009-04-20 20:36:00
Python入门教程(十四)Python的集合
2022-11-29 21:21:09
windows 10下安装搭建django1.10.3和Apache2.4的方法
2022-01-01 01:07:51
制作主页的独门功夫五十招
2010-09-05 21:15:00
php下实现伪 url 的超简单方法[转]
2024-05-02 17:08:18
Python Pandas中布尔索引的用法详解
2023-11-04 08:37:45
解决Mysql5.7.17在windows下安装启动时提示不成功问题
2024-01-16 07:39:14
django中静态文件配置static的方法
2022-07-29 08:52:51
详解Golang实现请求限流的几种办法
2024-04-26 17:19:16
IE6局部调用PNG32合并图片
2009-03-11 21:24:00
瀑布流布局浅析
2011-09-16 20:18:09
PythonPC客户端自动化实现原理(pywinauto)
2022-11-27 02:24:27
原生JS实现跑马灯效果
2024-04-19 09:53:11
mysql5.5与mysq 5.6中禁用innodb引擎的方法
2024-01-21 13:55:42
详解Python中time()方法的使用的教程
2022-10-07 16:28:23
Python-ElasticSearch搜索查询的讲解
2023-12-30 01:09:27
详解Pycharm与anaconda安装配置指南
2022-09-24 01:51:45
Java使用JDBC连接数据库
2024-01-22 03:21:45
img标签中alt和title属性的正确使用
2008-01-10 12:59:00
oracle停止数据库后linux完全卸载oracle的详细步骤
2024-01-20 12:10:20