PyTorch加載預(yù)訓(xùn)練模型實例(pretrained)
更新時間：2020年01月17日 08:46:14 作者：Xie_learning
今天小編就為大家分享一篇PyTorch加載預(yù)訓(xùn)練模型實例(pretrained)，具有很好的參考價值，希望對大家有所幫助。一起跟隨小編過來看看吧
使用預(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)練模型實例(pretrained)就是小編分享給大家的全部內(nèi)容了，希望能給大家一個參考，也希望大家多多支持腳本之家。
相關(guān)文章
python自定義函數(shù)實現(xiàn)一個數(shù)的三次方計算方法
今天小編就為大家分享一篇python自定義函數(shù)實現(xiàn)一個數(shù)的三次方計算方法，具有很好的參考價值，希望對大家有所幫助。一起跟隨小編過來看看吧2019-01-01
這篇文章主要為大家詳細介紹了python如何爬取m3u8連接的視頻，具有一定的參考價值，感興趣的小伙伴們可以參考一下2018-02-02
這篇文章主要介紹了pyinstaller打包后,配置文件無法正常讀取的解決方案，具有很好的參考價值，希望對大家有所幫助。如有錯誤或未考慮完全的地方，望不吝賜教2022-02-02
Python+Selenium實現(xiàn)在Geoserver批量發(fā)布Mongo矢量數(shù)據(jù)
這篇文章主要為大家詳細介紹了如何利用Python+Selenium實現(xiàn)在 Geoserver批量發(fā)布來自Mongo中的矢量數(shù)據(jù)，文中的示例代碼講解詳細，感興趣的小伙伴可以了解一下2022-07-07
這篇文章主要介紹了python中管道用法,實例分析了管道的創(chuàng)建、輸入、輸出及關(guān)閉等技巧,需要的朋友可以參考下2015-06-06
queue模塊是Python內(nèi)置的標(biāo)準(zhǔn)模塊，可以直接通過import?queue引用，這篇文章主要介紹了Python?queue模塊都具有哪些功能,需要的朋友可以參考下2023-04-04
一幅圖片，想通過程序判斷獲得其主要色調(diào)，應(yīng)該怎么樣處理？本文通過python實現(xiàn)判斷、獲取一張圖片的主色調(diào)方法,需要的朋友可以參考下2014-04-04
應(yīng)用OpenCV和Python進行SIFT算法的實現(xiàn)詳解
這篇文章主要介紹了應(yīng)用OpenCV和Python進行SIFT算法的實現(xiàn)詳解，文中通過示例代碼介紹的非常詳細，對大家的學(xué)習(xí)或者工作具有一定的參考學(xué)習(xí)價值，需要的朋友們下面隨著小編來一起學(xué)習(xí)學(xué)習(xí)吧2019-08-08
python實現(xiàn)一次性封裝多條sql語句(begin end)
這篇文章主要介紹了python實現(xiàn)一次性封裝多條sql語句(begin end)，具有很好的參考價值，希望對大家有所幫助。一起跟隨小編過來看看吧2020-06-06
Queue是python標(biāo)準(zhǔn)庫中的線程安全的隊列（FIFO）實現(xiàn),提供了一個適用于多線程編程的先進先出的數(shù)據(jù)結(jié)構(gòu)，即隊列，用來在生產(chǎn)者和消費者線程之間的信息傳遞，本文給大家講解Python隊列Queue的相關(guān)知識，需要的朋友參考下吧2023-04-04