快捷導(dǎo)航

Pytorch修改ResNet模型全連接層進(jìn)行直接訓(xùn)練實(shí)例

更新時(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)練時(shí)，是固定除最后一層的前面層權(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)式經(jīng)典代碼
這篇文章主要介紹了python列表推導(dǎo)式，文章內(nèi)容附有經(jīng)典的詳細(xì)代碼，感興趣的小伙按可以參考下面文章具體內(nèi)容
2021-10-10
PyQt5實(shí)現(xiàn)暗黑風(fēng)格的計(jì)時(shí)器
這篇文章主要為大家詳細(xì)介紹了PyQt5實(shí)現(xiàn)暗黑風(fēng)格的計(jì)時(shí)器，具有一定的參考價(jià)值，感興趣的小伙伴們可以參考一下
2019-07-07
使用paramiko遠(yuǎn)程執(zhí)行命令、下發(fā)文件的實(shí)例
下面小編就為大家?guī)硪黄褂胮aramiko遠(yuǎn)程執(zhí)行命令、下發(fā)文件的實(shí)例。小編覺得挺不錯(cuò)的，現(xiàn)在就分享給大家，也給大家做個(gè)參考。一起跟隨小編過來看看吧
2017-10-10
Python字符串格式化之使用format()方法
這篇文章主要介紹了Python字符串格式化之使用format()方法,具有很好的參考價(jià)值,希望對大家有所幫助,如有錯(cuò)誤或未考慮完全的地方,望不吝賜教
2024-08-08
Python常駐任務(wù)實(shí)現(xiàn)接收外界參數(shù)代碼解析
這篇文章主要介紹了Python常駐任務(wù)實(shí)現(xiàn)接收外界參數(shù)代碼解析,文中通過示例代碼介紹的非常詳細(xì)，對大家的學(xué)習(xí)或者工作具有一定的參考學(xué)習(xí)價(jià)值,需要的朋友可以參考下
2020-07-07
Python讀取文件的四種方式的實(shí)例詳解
這篇文章將從做核酸采集這一實(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
python使用ddddocr庫識別滑動(dòng)驗(yàn)證碼簡單代碼示例
這篇文章主要介紹了如何使用ddddocr庫來識別滑塊驗(yàn)證碼,并提供了一個(gè)示例代碼和識別結(jié)果,同時(shí)提醒注意ddddocr庫的大小限制,可能會影響某些無服務(wù)器函數(shù)的部署,需要的朋友可以參考下
2024-11-11
解決python3在anaconda下安裝caffe失敗的問題
下面小編就為大家?guī)硪黄鉀Qpython3在anaconda下安裝caffe失敗的問題。小編覺得挺不錯(cuò)的，現(xiàn)在就分享給大家，也給大家做個(gè)參考。一起跟隨小編過來看看吧
2017-06-06
python爬蟲多次請求超時(shí)的幾種重試方法(6種)
這篇文章主要介紹了python爬蟲多次請求超時(shí)的幾種重試方法，文中通過示例代碼介紹的非常詳細(xì)，對大家的學(xué)習(xí)或者工作具有一定的參考學(xué)習(xí)價(jià)值，需要的朋友們下面隨著小編來一起學(xué)習(xí)學(xué)習(xí)吧
2020-12-12