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函數是一個非常強大的函數,它可以將字符串作為代碼進行解析和執(zhí)行,本文主要介紹了如何使用eval函數解析和執(zhí)行字符串,需要的可以了解下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