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基础的相关知识,文章围绕着Python文件读写展开,文中有非常详细的介绍及代码示例,需要的朋友可以参考下2021-06-06
这篇文章主要介绍了python 模拟贷款卡号生成规则过程解析,文中通过示例代码介绍的非常详细，对大家的学习或者工作具有一定的参考学习价值,需要的朋友可以参考下2019-08-08
这篇文章主要介绍了Python如何优雅删除字符列表空字符及None元素,文中通过示例代码介绍的非常详细，对大家的学习或者工作具有一定的参考学习价值,需要的朋友可以参考下2020-06-06
这篇文章主要介绍了Python+Opencv实现数字识别的示例代码，文中通过示例代码介绍的非常详细，对大家的学习或者工作具有一定的参考学习价值，需要的朋友们下面随着小编来一起学习学习吧2021-03-03
这篇文章主要介绍了Python实现简单的列表冒泡排序和反转列表操作,涉及Python列表遍历、排序、追加等相关操作技巧,需要的朋友可以参考下2019-07-07
这篇文章主要介绍了Python删除字符串中字符的四种方法,本文通过示例代码给大家介绍的非常详细，对大家的学习或工作具有一定的参考借鉴价值，需要的朋友可以参考下2022-12-12
使用 Python 和 Selenium 解决 Cloudflare&
Cloudflare 验证码是一种用于区分人类用户和自动化机器人的功能,它是 Cloudflare 安全服务的重要组成部分,旨在防御网站免受自动化攻击和滥用,这篇文章主要介绍了使用 Python 和 Selenium 解决 Cloudflare 验证码,需要的朋友可以参考下2024-06-06
这篇文章主要介绍了Python生成不重复随机值的方法,实例分析了Python算法实现与Python自带方法的实现技巧,非常简单实用,需要的朋友可以参考下2015-05-05
shutil是 python 中的高级文件操作模块，与os模块形成互补的关系，os主要提供了文件或文件夹的新建、删除、查看等方法，还提供了对文件以及目录的路径操作，这篇文章主要介绍了python包合集-shutil,需要的朋友可以参考下2022-08-08
这篇文章主要介绍了自动化Nginx服务器的反向代理的配置方法,反向代理是Nginx服务器的招牌功能,需要的朋友可以参考下2015-06-06
最新评论