Pytorch修改ResNet模型全連接層進(jìn)行直接訓(xùn)練實(shí)例
更新時間：2019年09月10日 16:32:40 作者：嶙羽
在本篇文章里小編給大家整理的是關(guān)于Pytorch修改ResNet模型全連接層進(jìn)行直接訓(xùn)練相關(guān)知識點(diǎn)，有需要的朋友們參考下。
之前在用預(yù)訓(xùn)練的ResNet的模型進(jìn)行遷移訓(xùn)練時，是固定除最后一層的前面層權(quán)重，然后把全連接層輸出改為自己需要的數(shù)目，進(jìn)行最后一層的訓(xùn)練，那么現(xiàn)在假如想要只是把
最后一層的輸出改一下，不需要加載前面層的權(quán)重，方法如下：
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)
首先模型結(jié)構(gòu)是必須要傳入的，然后把最后一層的輸出改為自己所需的數(shù)目
以上知識點(diǎn)很簡單，大家可以測試下，感謝大家的閱讀和對腳本之家的支持。
相關(guān)文章
這篇文章主要介紹了python列表推導(dǎo)式，文章內(nèi)容附有經(jīng)典的詳細(xì)代碼，感興趣的小伙按可以參考下面文章具體內(nèi)容2021-10-10
PyQt5實(shí)現(xiàn)暗黑風(fēng)格的計時器
這篇文章主要為大家詳細(xì)介紹了PyQt5實(shí)現(xiàn)暗黑風(fēng)格的計時器，具有一定的參考價值，感興趣的小伙伴們可以參考一下2019-07-07
使用paramiko遠(yuǎn)程執(zhí)行命令、下發(fā)文件的實(shí)例
下面小編就為大家?guī)硪黄褂胮aramiko遠(yuǎn)程執(zhí)行命令、下發(fā)文件的實(shí)例。小編覺得挺不錯的，現(xiàn)在就分享給大家，也給大家做個參考。一起跟隨小編過來看看吧2017-10-10
這篇文章主要介紹了Python字符串格式化之使用format()方法,具有很好的參考價值,希望對大家有所幫助,如有錯誤或未考慮完全的地方,望不吝賜教2024-08-08
Python常駐任務(wù)實(shí)現(xiàn)接收外界參數(shù)代碼解析
這篇文章主要介紹了Python常駐任務(wù)實(shí)現(xiàn)接收外界參數(shù)代碼解析,文中通過示例代碼介紹的非常詳細(xì)，對大家的學(xué)習(xí)或者工作具有一定的參考學(xué)習(xí)價值,需要的朋友可以參考下2020-07-07
這篇文章將從做核酸采集這一實(shí)際問題出發(fā)，詳細(xì)講解Python讀取文件的四種方式，文中的示例代碼講解詳細(xì)，感興趣的小伙伴可以了解一下2022-03-03
python使用tkinter實(shí)現(xiàn)自定義多參數(shù)對話框
Tkinter模塊是Python標(biāo)準(zhǔn)庫中的一部分,用于創(chuàng)建圖形用戶界面（GUI）應(yīng)用程序,它提供了一組工具和組件,用于創(chuàng)建窗口、按鈕、文本框等用戶界面元素,并且可以響應(yīng)用戶的輸入,本文將給大家講講python如何使用tkinter實(shí)現(xiàn)自定義多參數(shù)對話框2023-08-08
這篇文章主要介紹了如何使用ddddocr庫來識別滑塊驗證碼,并提供了一個示例代碼和識別結(jié)果,同時提醒注意ddddocr庫的大小限制,可能會影響某些無服務(wù)器函數(shù)的部署,需要的朋友可以參考下2024-11-11
解決python3在anaconda下安裝caffe失敗的問題
下面小編就為大家?guī)硪黄鉀Qpython3在anaconda下安裝caffe失敗的問題。小編覺得挺不錯的，現(xiàn)在就分享給大家，也給大家做個參考。一起跟隨小編過來看看吧2017-06-06
這篇文章主要介紹了python爬蟲多次請求超時的幾種重試方法，文中通過示例代碼介紹的非常詳細(xì)，對大家的學(xué)習(xí)或者工作具有一定的參考學(xué)習(xí)價值，需要的朋友們下面隨著小編來一起學(xué)習(xí)學(xué)習(xí)吧2020-12-12