pytorch 修改預(yù)訓練model實例
更新時間:2020年01月18日 15:38:52 作者:whut_ldz
今天小編就為大家分享一篇pytorch 修改預(yù)訓練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ù)訓練model實例就是小編分享給大家的全部內(nèi)容了,希望能給大家一個參考,也希望大家多多支持腳本之家。
您可能感興趣的文章:
- 聊聊pytorch測試的時候為何要加上model.eval()
- pytorch中的model.eval()和BN層的使用
- 解決Pytorch中的神坑:關(guān)于model.eval的問題
- Pytorch BertModel的使用說明
- PyTorch中model.zero_grad()和optimizer.zero_grad()用法
- pytorch掉坑記錄:model.eval的作用說明
- pytorch:model.train和model.eval用法及區(qū)別詳解
- pytorch查看torch.Tensor和model是否在CUDA上的實例
- pytorch中的model=model.to(device)使用說明
相關(guān)文章
LyScript實現(xiàn)計算片段Hash并寫出Excel的示例代碼
本案例將學習運用LyScript計算特定程序中特定某些片段的Hash特征值,并通過xlsxwriter這個第三方模塊將計算到的hash值存儲成一個excel表格,感興趣的可以跟隨小編一起學習一下2022-09-09Python編寫的com組件發(fā)生R6034錯誤的原因與解決辦法
pythoncom27.dll可能沒有包含manifest信息,或者沒有包含正確的manifest信息,或者系統(tǒng)中的c++ runtime library受到破壞都有可能造成這種現(xiàn)象2013-04-04基于Python函數(shù)的作用域規(guī)則和閉包(詳解)
下面小編就為大家分享一篇基于Python函數(shù)的作用域規(guī)則和閉包詳解,希望對大家有所幫助。一起跟隨小編過來看看吧2017-11-11詳解Python調(diào)用華為API實現(xiàn)圖像標簽
華為云圖像標簽可識別上千種通用物體以及數(shù)百種場景標簽,一個圖像可包含多個標簽內(nèi)容,語義內(nèi)容非常豐富。本文將通過Python調(diào)用華為API實現(xiàn)圖像標簽,需要的可以參考一下2022-04-04