快捷導(dǎo)航

pytorch 修改預(yù)訓(xùn)練model實例

更新時間：2020年01月18日 15:38:52 作者：whut_ldz

今天小編就為大家分享一篇pytorch 修改預(yù)訓(xùn)練model實例，具有很好的參考價值，希望對大家有所幫助。一起跟隨小編過來看看吧

我就廢話不多說了，直接上代碼吧！

 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

resnet = models.resnet50(pretrained=True)
model = Net(resnet)

以上這篇pytorch 修改預(yù)訓(xùn)練model實例就是小編分享給大家的全部內(nèi)容了，希望能給大家一個參考，也希望大家多多支持腳本之家。

您可能感興趣的文章:

相關(guān)文章

Python使用自帶的base64庫進(jìn)行base64編碼和解碼
在Python中,處理數(shù)據(jù)的編碼和解碼是數(shù)據(jù)傳輸和存儲中非常普遍的需求,其中,Base64是一種常用的編碼方案,本文我將詳細(xì)介紹如何使用Python的base64庫進(jìn)行Base64編碼和解碼,需要的朋友可以參考下
2025-04-04
用Cython加速Python到“起飛”(推薦)
這篇文章主要介紹了用Cython加速Python到“起飛”，文中通過示例代碼介紹的非常詳細(xì)，對大家的學(xué)習(xí)或者工作具有一定的參考學(xué)習(xí)價值，需要的朋友們下面隨著小編來一起學(xué)習(xí)學(xué)習(xí)吧
2019-08-08
Python基于execjs運行js過程解析
這篇文章主要介紹了Python基于execjs運行js過程解析,文中通過示例代碼介紹的非常詳細(xì)，對大家的學(xué)習(xí)或者工作具有一定的參考學(xué)習(xí)價值,需要的朋友可以參考下
2020-11-11
python 寫函數(shù)在一定條件下需要調(diào)用自身時的寫法說明
這篇文章主要介紹了python 寫函數(shù)在一定條件下需要調(diào)用自身時的寫法說明，具有很好的參考價值，希望對大家有所幫助。一起跟隨小編過來看看吧
2020-06-06
python把轉(zhuǎn)列表為集合的方法
在本篇文章里小編給大家分享了關(guān)于python把轉(zhuǎn)列表為集合的方法以及相關(guān)實例內(nèi)容，有興趣的朋友們學(xué)習(xí)下。
2019-06-06
Python基于Tkinter的HelloWorld入門實例
這篇文章主要介紹了Python基于Tkinter的HelloWorld入門實例,以一個簡單實例分析了Python中Tkinter模塊的使用技巧,需要的朋友可以參考下
2015-06-06
python Socket網(wǎng)絡(luò)編程實現(xiàn)C/S模式和P2P
這篇文章主要介紹了python Socket網(wǎng)絡(luò)編程實現(xiàn)C/S模式和P2P，文中通過示例代碼介紹的非常詳細(xì)，對大家的學(xué)習(xí)或者工作具有一定的參考學(xué)習(xí)價值，需要的朋友們下面隨著小編來一起學(xué)習(xí)學(xué)習(xí)吧
2020-06-06
python實現(xiàn)摳圖給證件照換背景源碼
這篇文章主要為大家詳細(xì)介紹了python實現(xiàn)摳圖給證件照換背景源碼，具有一定的參考價值，感興趣的小伙伴們可以參考一下
2019-08-08
python Django的web開發(fā)實例（入門）
這篇文章主要介紹了python Django的web開發(fā)實例，文中通過示例代碼介紹的非常詳細(xì)，對大家的學(xué)習(xí)或者工作具有一定的參考學(xué)習(xí)價值，需要的朋友們下面隨著小編來一起學(xué)習(xí)學(xué)習(xí)吧
2019-07-07
Python 查詢學(xué)生的平均分?jǐn)?shù)的實現(xiàn)
這篇文章主要為大家詳細(xì)介紹了Python 查詢學(xué)生的平均分?jǐn)?shù)的實現(xiàn),文中示例代碼介紹的非常詳細(xì),具有一定的參考價值,感興趣的小伙伴們可以參考一下
2019-04-04