PyTorch加載預(yù)訓(xùn)練模型實(shí)例(pretrained)
更新時(shí)間：2020年01月17日 08:46:14 作者：Xie_learning
今天小編就為大家分享一篇PyTorch加載預(yù)訓(xùn)練模型實(shí)例(pretrained)，具有很好的參考價(jià)值，希望對大家有所幫助。一起跟隨小編過來看看吧
使用預(yù)訓(xùn)練模型的代碼如下：
# 加載預(yù)訓(xùn)練模型
resNet50 = models.resnet50(pretrained=True)
ResNet50 = ResNet(Bottleneck, [3, 4, 6, 3], num_classes=2)
# 讀取參數(shù)
pretrained_dict = resNet50.state_dict()
model_dict = ResNet50.state_dict()
# 將pretained_dict里不屬于model_dict的鍵剔除掉
pretrained_dict = {k: v for k, v in pretrained_dict.items() if k in model_dict}
# 更新現(xiàn)有的model_dict
model_dict.update(pretrained_dict)
# 加載真正需要的state_dict
ResNet50.load_state_dict(model_dict)
以上這篇PyTorch加載預(yù)訓(xùn)練模型實(shí)例(pretrained)就是小編分享給大家的全部內(nèi)容了，希望能給大家一個(gè)參考，也希望大家多多支持腳本之家。
相關(guān)文章
利用Pandas實(shí)現(xiàn)對數(shù)據(jù)進(jìn)行移動(dòng)計(jì)算
這篇文章主要為大家詳細(xì)介紹了如何利用Pandas實(shí)現(xiàn)對數(shù)據(jù)進(jìn)行移動(dòng)計(jì)算，文中的示例代碼講解詳細(xì)，對我們了解Pandas有一定幫助，需要的可以參考一下2022-07-07
Python實(shí)現(xiàn)好友全頭像的拼接實(shí)例(推薦)
下面小編就為大家?guī)硪黄狿ython實(shí)現(xiàn)好友全頭像的拼接實(shí)例(推薦)。小編覺得挺不錯(cuò)的，現(xiàn)在就分享給大家，也給大家做個(gè)參考。一起跟隨小編過來看看吧2017-06-06
Python入門教程(十九)python的函數(shù)詳解
這篇文章主要介紹了Python入門教程(十九)python的函數(shù),函數(shù)是組織好的，可重復(fù)使用的，用來實(shí)現(xiàn)單一，或相關(guān)聯(lián)功能的代碼段,需要的朋友可以參考下2023-04-04
今天小編就為大家分享一篇python 畫函數(shù)曲線示例，具有很好的參考價(jià)值，希望對大家有所幫助。一起跟隨小編過來看看吧2019-12-12
今天小編就為大家分享一篇python交互界面的退出方法，具有很好的參考價(jià)值，希望對大家有所幫助。一起跟隨小編過來看看吧2019-02-02
Python使用Flask框架同時(shí)上傳多個(gè)文件的方法
這篇文章主要介紹了Python使用Flask框架同時(shí)上傳多個(gè)文件的方法,實(shí)例分析了Python中Flask框架操作文件實(shí)現(xiàn)上傳的技巧,需要的朋友可以參考下2015-03-03
python 同時(shí)運(yùn)行多個(gè)程序的實(shí)例
今天小編就為大家分享一篇python 同時(shí)運(yùn)行多個(gè)程序的實(shí)例，具有很好的參考價(jià)值，希望對大家有所幫助。一起跟隨小編過來看看吧2019-01-01
python爬蟲爬取bilibili網(wǎng)頁基本內(nèi)容
這篇文章主要介紹了python爬蟲爬取bilibili網(wǎng)頁基本內(nèi)容，用爬蟲爬取bilibili網(wǎng)站排行榜游戲類的所有名稱及鏈接，下面來看看具體的實(shí)現(xiàn)過程吧,需要的朋友可以參考一下2022-01-01
在Python中使用PIL模塊對圖片進(jìn)行高斯模糊處理的教程
這篇文章主要介紹了在Python中使用PIL模塊對圖片進(jìn)行高斯模糊處理的教程,這個(gè)無圖形界面的腳本代碼非常簡單,需要的朋友可以參考下2015-05-05
Python使用selenium實(shí)現(xiàn)網(wǎng)頁用戶名 密碼 驗(yàn)證碼自動(dòng)登錄功能
這篇文章主要介紹了Python使用selenium實(shí)現(xiàn)網(wǎng)頁用戶名 密碼 驗(yàn)證碼自動(dòng)登錄功能，實(shí)現(xiàn)思路很簡單，感興趣的朋友跟隨腳本之家小編一起學(xué)習(xí)吧2018-05-05