快捷導(dǎo)航

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

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

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

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

 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實(shí)例就是小編分享給大家的全部?jī)?nèi)容了，希望能給大家一個(gè)參考，也希望大家多多支持腳本之家。

您可能感興趣的文章:

相關(guān)文章

Pygame游戲開(kāi)發(fā)實(shí)例講解之圖形繪制與鍵鼠事件
這篇文章主要介紹了Pygame游戲開(kāi)發(fā)中常用的圖形繪制與鍵鼠事件實(shí)例，文中通過(guò)示例代碼介紹的非常詳細(xì)，對(duì)大家的學(xué)習(xí)或者工作具有一定的參考學(xué)習(xí)價(jià)值，需要的朋友們下面隨著小編來(lái)一起學(xué)習(xí)吧
2022-11-11
Python3.6 之后字典是有序的?
字典數(shù)據(jù)是有序的, 但是這個(gè)序不是由外部控制, 而是內(nèi)部字典定位機(jī)制的序所以對(duì)外來(lái)講, 數(shù)據(jù)本身是無(wú)序的你每次遍歷的順序一樣, 是因?yàn)槊杜e結(jié)果是按內(nèi)部排序輸出而無(wú)序則表示在你無(wú)法從外部控制最終的輸出順序，下面我們來(lái)學(xué)習(xí)Python字典有序性的相關(guān)資料又當(dāng)怎樣吧
2021-12-12
Python實(shí)現(xiàn)TCP通信的示例代碼
這篇文章主要介紹了Python實(shí)現(xiàn)TCP通信的示例代碼，文中通過(guò)示例代碼介紹的非常詳細(xì)，對(duì)大家的學(xué)習(xí)或者工作具有一定的參考學(xué)習(xí)價(jià)值，需要的朋友們下面隨著小編來(lái)一起學(xué)習(xí)學(xué)習(xí)吧
2019-09-09
LyScript實(shí)現(xiàn)計(jì)算片段Hash并寫(xiě)出Excel的示例代碼
本案例將學(xué)習(xí)運(yùn)用LyScript計(jì)算特定程序中特定某些片段的Hash特征值，并通過(guò)xlsxwriter這個(gè)第三方模塊將計(jì)算到的hash值存儲(chǔ)成一個(gè)excel表格，感興趣的可以跟隨小編一起學(xué)習(xí)一下
2022-09-09
Python爬蟲(chóng)之正則表達(dá)式基本用法實(shí)例分析
這篇文章主要介紹了Python爬蟲(chóng)之正則表達(dá)式基本用法,結(jié)合實(shí)例形式分析了Python正則表達(dá)式的基本概念、函數(shù)語(yǔ)法、相關(guān)使用方法及操作注意事項(xiàng),需要的朋友可以參考下
2018-08-08
Python編寫(xiě)的com組件發(fā)生R6034錯(cuò)誤的原因與解決辦法
pythoncom27.dll可能沒(méi)有包含manifest信息，或者沒(méi)有包含正確的manifest信息，或者系統(tǒng)中的c++ runtime library受到破壞都有可能造成這種現(xiàn)象
2013-04-04
基于Python函數(shù)的作用域規(guī)則和閉包(詳解)
下面小編就為大家分享一篇基于Python函數(shù)的作用域規(guī)則和閉包詳解，希望對(duì)大家有所幫助。一起跟隨小編過(guò)來(lái)看看吧
2017-11-11
python中List的sort方法指南
我們需要對(duì)List進(jìn)行排序，Python提供了兩個(gè)方法:1.用List的成員函數(shù)sort進(jìn)行排序;2.用built-in函數(shù)sorted進(jìn)行排序,今天我們就來(lái)探討下這2個(gè)方法
2014-09-09
Python算術(shù)運(yùn)算符實(shí)例詳解
這篇文章主要介紹了Python算術(shù)運(yùn)算符實(shí)例詳解的相關(guān)資料,需要的朋友可以參考下
2017-05-05
詳解Python調(diào)用華為API實(shí)現(xiàn)圖像標(biāo)簽
華為云圖像標(biāo)簽可識(shí)別上千種通用物體以及數(shù)百種場(chǎng)景標(biāo)簽，一個(gè)圖像可包含多個(gè)標(biāo)簽內(nèi)容，語(yǔ)義內(nèi)容非常豐富。本文將通過(guò)Python調(diào)用華為API實(shí)現(xiàn)圖像標(biāo)簽，需要的可以參考一下
2022-04-04