pytorch 更改預(yù)訓(xùn)練模型網(wǎng)絡(luò)結(jié)構(gòu)的方法
更新時間：2019年08月19日 14:56:03 作者：wayne980
今天小編就為大家分享一篇pytorch 更改預(yù)訓(xùn)練模型網(wǎng)絡(luò)結(jié)構(gòu)的方法，具有很好的參考價值，希望對大家有所幫助。一起跟隨小編過來看看吧
一個繼承nn.module的model它包含一個叫做children()的函數(shù)，這個函數(shù)可以用來提取出model每一層的網(wǎng)絡(luò)結(jié)構(gòu)，在此基礎(chǔ)上進(jìn)行修改即可，修改方法如下(去除后兩層)：
resnet_layer = nn.Sequential(*list(model.children())[:-2])
那么，接下來就可以構(gòu)建我們的網(wǎng)絡(luò)了：
class Net(nn.Module):
def __init__(self , model):
super(Net, self).__init__()
#取掉model的后兩層
self.resnet_layer = nn.Sequential(*list(model.children())[:-2])
self.transion_layer = nn.ConvTranspose2d(2048, 2048, kernel_size=14, stride=3)
self.pool_layer = nn.MaxPool2d(32)
self.Linear_layer = nn.Linear(2048, 8)
def forward(self, x):
x = self.resnet_layer(x)
x = self.transion_layer(x)
x = self.pool_layer(x)
x = x.view(x.size(0), -1)
x = self.Linear_layer(x)
return x
最后，構(gòu)建一個對象，并加載resnet預(yù)訓(xùn)練的參數(shù)就可以啦～
resnet = models.resnet50(pretrained=True) model = Net(resnet)
以上這篇pytorch 更改預(yù)訓(xùn)練模型網(wǎng)絡(luò)結(jié)構(gòu)的方法就是小編分享給大家的全部內(nèi)容了，希望能給大家一個參考，也希望大家多多支持腳本之家。
相關(guān)文章
python深度學(xué)習(xí)借助多標(biāo)簽分類器進(jìn)行對抗訓(xùn)練
這篇文章主要為大家介紹了python深度學(xué)習(xí)中如何借助多標(biāo)簽分類器進(jìn)行對抗訓(xùn)練，有需要的朋友可以借鑒參考下，希望能夠有所幫助，祝大家多多進(jìn)步2022-01-01
Python在報表自動化的優(yōu)勢及實現(xiàn)流程
本文利用Python實現(xiàn)報表自動化,通過介紹環(huán)境設(shè)置、數(shù)據(jù)收集和準(zhǔn)備、報表生成以及自動化流程,展示Python的靈活性和豐富的生態(tài)系統(tǒng)在報表自動化中的卓越表現(xiàn),從設(shè)置虛擬環(huán)境到使用Pandas和Matplotlib處理數(shù)據(jù),到借助APScheduler實現(xiàn)定期自動化,每個步驟都得到詳盡闡述2023-12-12
這篇文章主要介紹了django之如何按日期查詢數(shù)據(jù)問題,具有很好的參考價值,希望對大家有所幫助,如有錯誤或未考慮完全的地方,望不吝賜教2023-08-08
這篇文章主要為大家介紹了Python項目管理工具Poetry使用示例詳解，有需要的朋友可以借鑒參考下，希望能夠有所幫助，祝大家多多進(jìn)步，早日升職加薪2023-09-09
這篇文章主要介紹了python?多線程threading程序詳情，Thread類表示在單獨的控制線程中運行的活動，下文相關(guān)資料需要的小伙伴可以參考一下2022-04-04
這篇文章主要為大家詳細(xì)介紹了python實現(xiàn)上傳下載文件功能，文中示例代碼介紹的非常詳細(xì)，具有一定的參考價值，感興趣的小伙伴們可以參考一下2017-07-07
這篇文章主要介紹了如何在Python?中獲取單成員集合中的唯一元素,本文給大家介紹的非常詳細(xì)，對大家的學(xué)習(xí)或工作具有一定的參考借鑒價值，需要的朋友可以參考下2023-03-03
Python實現(xiàn)RabbitMQ6種消息模型的示例代碼
這篇文章主要介紹了Python實現(xiàn)RabbitMQ6種消息模型的示例代碼，文中通過示例代碼介紹的非常詳細(xì)，對大家的學(xué)習(xí)或者工作具有一定的參考學(xué)習(xí)價值，需要的朋友們下面隨著小編來一起學(xué)習(xí)學(xué)習(xí)吧2020-03-03
在本篇文章里我們給大家分享關(guān)于IntelliJ IDEA安裝運行python插件方法，對此有需求的讀者們可以跟著步驟學(xué)習(xí)下2018-12-12
這篇文章主要介紹了Python搭建HTTP服務(wù)過程圖解,文中通過示例代碼介紹的非常詳細(xì)，對大家的學(xué)習(xí)或者工作具有一定的參考學(xué)習(xí)價值,需要的朋友可以參考下2019-12-12