pytorch實現(xiàn)從本地加載 .pth 格式模型
更新時間：2020年02月14日 11:51:34 作者：盧開毅
今天小編就為大家分享一篇pytorch實現(xiàn)從本地加載 .pth 格式模型，具有很好的參考價值，希望對大家有所幫助。一起跟隨小編過來看看吧
可以從官網(wǎng)加載預(yù)訓(xùn)練好的模型：
import torchvision.models as models model = models.vgg16(pretrained = True) print(model)
但是經(jīng)常會出現(xiàn)因為下載速度太慢而出現(xiàn)requests.exceptions.ConnectionError: ('Connection aborted.', TimeoutError(10060, '由于連接方在一段時間后沒有正確答復(fù)或連接的主機沒有反應(yīng)，連接嘗試失敗。', None, 10060, None))這種錯誤，因此需要我們手動去下載 .pth 文件（百度云也很慢，如果你是SVIP，當(dāng)我沒說；迅雷的速度也還可以），然后從本地加載。
從本地加載只需要把上面的代碼換成如下：
import torchvision.models as models model = models.vgg16(pretrained=False) pre=torch.load(r'.\kaggle_dog_vs_cat\pretrain\vgg16-397923af.pth') model.load_state_dict(pre)
如果你模型不是用的vgg16，而是用的vgg11或者vgg13，只需要修改語句 model = models.vgg16(pretrained=False) 為對應(yīng)模型的函數(shù)即可。
以上這篇pytorch實現(xiàn)從本地加載 .pth 格式模型就是小編分享給大家的全部內(nèi)容了，希望能給大家一個參考，也希望大家多多支持腳本之家。
相關(guān)文章
這篇文章主要介紹了python tkinter事件綁定，文中通過示例代碼介紹的非常詳細(xì)，對大家的學(xué)習(xí)或者工作具有一定的參考學(xué)習(xí)價值，需要的朋友們下面隨著小編來一起學(xué)習(xí)學(xué)習(xí)吧2019-03-03
今天小編就為大家分享一篇pandas 根據(jù)列的值選取所有行的示例，具有很好的參考價值，希望對大家有所幫助。一起跟隨小編過來看看吧2018-11-11
使用Python3實現(xiàn)判斷函數(shù)的圈復(fù)雜度
編寫函數(shù)最重要的原則就是：別寫太復(fù)雜的函數(shù),那什么樣的函數(shù)才能算是過于復(fù)雜？一般會通過兩個標(biāo)準(zhǔn)來判斷,長度和圈復(fù)雜度,下面我們就來看看如何使用Python判斷函數(shù)的圈復(fù)雜度吧2024-04-04
Python的datetime模塊是處理日期和時間的強大工具,datetime類可以獲取當(dāng)前時間、指定日期、計算時間差、訪問時間屬性及格式化時間,這些功能使得在Python中進行時間日期處理變得簡單高效,需要的朋友可以參考下2024-09-09
這篇文章主要介紹了python實現(xiàn)三次密碼驗證的示例，幫助大家更好的理解和學(xué)習(xí)使用python，感興趣的朋友可以了解下2021-04-04
Django Rest framework認(rèn)證組件詳細(xì)用法
這篇文章主要介紹了Django Rest framework認(rèn)證組件詳細(xì)用法,文中通過示例代碼介紹的非常詳細(xì)，對大家的學(xué)習(xí)或者工作具有一定的參考學(xué)習(xí)價值,需要的朋友可以參考下2019-07-07
Python中Matplotlib圖像添加標(biāo)簽的方法實現(xiàn)
本文主要介紹了Python中Matplotlib圖像添加標(biāo)簽的方法實現(xiàn)，文中通過示例代碼介紹的非常詳細(xì)，對大家的學(xué)習(xí)或者工作具有一定的參考學(xué)習(xí)價值，需要的朋友們下面隨著小編來一起學(xué)習(xí)學(xué)習(xí)吧2023-04-04
Python??Pandas教程之使用?pandas.read_csv()?讀取?csv
這篇文章主要介紹了Python Pandas教程之使用pandas.read_csv()讀取csv，文章通過圍繞主題展開詳細(xì)的內(nèi)容介紹，具有一定的參考價值，需要的小伙伴可以參考一下2022-09-09
這篇文章主要為大家介紹了使用PyTorch常見4個錯誤解決示例詳解，有需要的朋友可以借鑒參考下，希望能夠有所幫助，祝大家多多進步，早日升職加薪2022-10-10
引用和復(fù)制是Python處理對象的重要概念,本文主要介紹了Python中對象的引用與復(fù)制的使用,文中通過示例代碼介紹的非常詳細(xì),對大家的學(xué)習(xí)或者工作具有一定的參考學(xué)習(xí)價值,需要的朋友們下面隨著小編來一起學(xué)習(xí)學(xué)習(xí)吧2023-12-12