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字典dict常用方法函数实例,文中通过示例代码介绍的非常详细，对大家的学习或者工作具有一定的参考学习价值,需要的朋友可以参考下2020-11-11
在Python中,eval函数是一个非常强大的函数,它可以将字符串作为代码进行解析和执行,本文主要介绍了如何使用eval函数解析和执行字符串,需要的可以了解下2024-01-01
这篇文章主要介绍了Python终端输出彩色字符方法详解,需要的朋友可以参考下2020-02-02
本文主要介绍了PyTorch中apex安装方式和避免踩坑，文中通过示例代码介绍的非常详细，对大家的学习或者工作具有一定的参考学习价值，需要的朋友们下面随着小编来一起学习学习吧2022-07-07
这篇文章将通过两个小问题：前N阶乘求和问题、金字塔数求和问题，带领大家深入浅出的理解两种语言的基本语法，并用以实际场景，需要的可以参考一下2022-03-03
因工作需要,要在图片中显示中文字符,并且要求速度足够快,在网上搜罗一番后,总结下几个解决方法,对python 解决OpenCV显示中文字符相关知识感兴趣的朋友一起看看吧2024-04-04
信号处理作为数字信号处理领域的关键技术,涵盖了从信号获取、传输、存储到最终应用的一系列处理步骤,在这篇博客中,我们将深入探讨Python中Scipy库在信号处理领域的应用,需要的朋友可以参考下2023-12-12
今天小编就为大家分享一篇浅谈Python3实现两个矩形的交并比(IoU)，具有很好的参考价值，希望对大家有所帮助。一起跟随小编过来看看吧2020-01-01
这篇文章主要为大家详细介绍了pytorch实现MNIST手写体识别，使用全连接神经网络，文中示例代码介绍的非常详细，具有一定的参考价值，感兴趣的小伙伴们可以参考一下2020-02-02
pandoc是一个强大的文档格式转换工具,支持丰富的格式转换,并尽可能的保留原来的排版,号称文档格式转换的瑞士军刀,本文将给大家介绍一下使用python搭配pandoc实现html批量转word,感兴趣的朋友可以参考阅读下2023-09-09
最新评论