Pytorch中的VGG实现修改最后一层FC
更新时间：2020年01月15日 08:36:22 作者：ForABiggerWorld
今天小编就为大家分享一篇Pytorch中的VGG实现修改最后一层FC，具有很好的参考价值，希望对大家有所帮助。一起跟随小编过来看看吧
https://discuss.pytorch.org/t/how-to-modify-the-final-fc-layer-based-on-the-torch-model/766/12
That's because vgg19 doesn't have a fc member variable. Instead, it has a
(classifier): Sequential ( (0): Dropout (p = 0.5) (1): Linear (25088 -> 4096) (2): ReLU (inplace) (3): Dropout (p = 0.5) (4): Linear (4096 -> 4096) (5): ReLU (inplace) (6): Linear (4096 -> 100) )
To replace the last linear layer, a temporary solution would be
vgg19.classifier._modules['6'] = nn.Linear(4096, 8)
以上这篇Pytorch中的VGG实现修改最后一层FC就是小编分享给大家的全部内容了，希望能给大家一个参考，也希望大家多多支持脚本之家。
您可能感兴趣的文章:
相关文章
这篇文章主要介绍了Python实现获取邮箱内容并解析的方法,结合完整实例形式分析了Python登陆pop3服务器并解析获取邮箱内容相关操作技巧,需要的朋友可以参考下2018-06-06
K-means算法是最为经典的基于划分的聚类方法,是十大经典数据挖掘算法之一,下面这篇文章主要给大家介绍了关于Python自定义指标聚类的相关资料,文中通过实例代码介绍的非常详细,需要的朋友可以参考下2022-02-02
在本篇文章里小编给大家整理的是一篇关于Python关于OS文件目录处理的实例内容，有兴趣的朋友们可以学一下。2021-05-05
这篇文章主要为大家详细介绍了如何使用python将一个文件分配到指定的多个文件夹,也就说将一个文件分配到一个母文件夹下的所有的子文件夹,感兴趣的可以了解下2025-01-01
这篇文章主要介绍了Python numpy数组转置与轴变换,文中通过示例代码介绍的非常详细，对大家的学习或者工作具有一定的参考学习价值,需要的朋友可以参考下2019-11-11
今天小编就为大家分享一篇python读取文本中的坐标方法，具有很好的参考价值，希望对大家有所帮助。一起跟随小编过来看看吧2018-10-10
这篇文章主要给大家介绍了关于Python进行图片验证码识别的相关资料,基于Python和OpenCV的验证码识别系统具有重要的研究意义和实际应用价值,文中通过代码介绍的非常详细,需要的朋友可以参考下2023-09-09
使用Python的Tornado框架实现一个简单的WebQQ机器人
这篇文章主要介绍了使用Python的Tornado框架实现一个简单的WebQQ机器人,Tornado的异步特性可以提高I/O性能,需要的朋友可以参考下2015-04-04
这篇文章主要介绍了TensorFlow2.0使用keras训练模型的实现，文中通过示例代码介绍的非常详细，对大家的学习或者工作具有一定的参考学习价值，需要的朋友们下面随着小编来一起学习学习吧2021-02-02
这篇文章主要介绍了Python3实现zip分卷压缩过程解析,文中通过示例代码介绍的非常详细，对大家的学习或者工作具有一定的参考学习价值,需要的朋友可以参考下2019-10-10
最新评论