Pytorch修改ResNet模型全连接层进行直接训练实例
更新时间：2019年09月10日 16:32:40 作者：嶙羽
在本篇文章里小编给大家整理的是关于Pytorch修改ResNet模型全连接层进行直接训练相关知识点，有需要的朋友们参考下。
之前在用预训练的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列表推导式，文章内容附有经典的详细代码，感兴趣的小伙按可以参考下面文章具体内容2021-10-10
这篇文章主要为大家详细介绍了PyQt5实现暗黑风格的计时器，具有一定的参考价值，感兴趣的小伙伴们可以参考一下2019-07-07
下面小编就为大家带来一篇使用paramiko远程执行命令、下发文件的实例。小编觉得挺不错的，现在就分享给大家，也给大家做个参考。一起跟随小编过来看看吧2017-10-10
这篇文章主要介绍了Python字符串格式化之使用format()方法,具有很好的参考价值,希望对大家有所帮助,如有错误或未考虑完全的地方,望不吝赐教2024-08-08
这篇文章主要介绍了Python常驻任务实现接收外界参数代码解析,文中通过示例代码介绍的非常详细，对大家的学习或者工作具有一定的参考学习价值,需要的朋友可以参考下2020-07-07
这篇文章将从做核酸采集这一实际问题出发，详细讲解Python读取文件的四种方式，文中的示例代码讲解详细，感兴趣的小伙伴可以了解一下2022-03-03
Tkinter模块是Python标准库中的一部分,用于创建图形用户界面（GUI）应用程序,它提供了一组工具和组件,用于创建窗口、按钮、文本框等用户界面元素,并且可以响应用户的输入,本文将给大家讲讲python如何使用tkinter实现自定义多参数对话框2023-08-08
这篇文章主要介绍了如何使用ddddocr库来识别滑块验证码,并提供了一个示例代码和识别结果,同时提醒注意ddddocr库的大小限制,可能会影响某些无服务器函数的部署,需要的朋友可以参考下2024-11-11
解决python3在anaconda下安装caffe失败的问题
下面小编就为大家带来一篇解决python3在anaconda下安装caffe失败的问题。小编觉得挺不错的，现在就分享给大家，也给大家做个参考。一起跟随小编过来看看吧2017-06-06
这篇文章主要介绍了python爬虫多次请求超时的几种重试方法，文中通过示例代码介绍的非常详细，对大家的学习或者工作具有一定的参考学习价值，需要的朋友们下面随着小编来一起学习学习吧2020-12-12
最新评论