pytorch實現從本地加載 .pth 格式模型
更新時間：2020年02月14日 11:51:34 作者：盧開毅
今天小編就為大家分享一篇pytorch實現從本地加載 .pth 格式模型，具有很好的參考價值，希望對大家有所幫助。一起跟隨小編過來看看吧
可以從官網加載預訓練好的模型：
import torchvision.models as models model = models.vgg16(pretrained = True) print(model)
但是經常會出現因為下載速度太慢而出現requests.exceptions.ConnectionError: ('Connection aborted.', TimeoutError(10060, '由于連接方在一段時間后沒有正確答復或連接的主機沒有反應，連接嘗試失敗。', None, 10060, None))這種錯誤，因此需要我們手動去下載 .pth 文件（百度云也很慢，如果你是SVIP，當我沒說；迅雷的速度也還可以），然后從本地加載。
從本地加載只需要把上面的代碼換成如下：
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) 為對應模型的函數即可。
以上這篇pytorch實現從本地加載 .pth 格式模型就是小編分享給大家的全部內容了，希望能給大家一個參考，也希望大家多多支持腳本之家。
相關文章
這篇文章主要介紹了Python 如何提高元組的可讀性,文中通過示例代碼介紹的非常詳細，對大家的學習或者工作具有一定的參考學習價值,需要的朋友可以參考下2019-08-08
這篇文章主要介紹了Django的models中on_delete參數詳解，文中通過示例代碼介紹的非常詳細，對大家的學習或者工作具有一定的參考學習價值，需要的朋友們下面隨著小編來一起學習學習吧2019-07-07
某些條件測試，需要一個簡單的web服務器測試一下，為此專門去配置個nginx 或者 apache服務器略顯麻煩,這里就為大家介紹一下使用python快速建立超簡單的web服務器的方法,需要的朋友可以參考下2018-02-02
今天小編就為大家分享一篇Python解決兩個整數相除只得到整數部分的實例，具有很好的參考價值，希望對大家有所幫助。一起跟隨小編過來看看吧2018-11-11
下面小編就為大家分享一篇用pandas按列合并兩個文件的實例，具有很好的參考價值，希望對大家有所幫助。一起跟隨小編過來看看吧2018-04-04
python中有一個zfill方法用來給字符串前面補0，非常不錯，下面小編給大家分享了實例代碼，非常不錯，具有參考借鑒價值，需要的朋友參考下吧2018-04-04
pyspark 讀取csv文件創(chuàng)建DataFrame的兩種方法
今天小編就為大家分享一篇pyspark 讀取csv文件創(chuàng)建DataFrame的兩種方法，具有很好的參考價值，希望對大家有所幫助。一起跟隨小編過來看看吧2018-06-06
這篇文章主要介紹了用Python爬取指定關鍵詞的微博，下面文章圍繞Python爬取指定關鍵詞的微博的相關資料展開詳細內容,需要的朋友可以參考一下2021-11-11
本文主要介紹了Python使用pandasai實現數據分析，文中通過示例代碼介紹的非常詳細，對大家的學習或者工作具有一定的參考學習價值，需要的朋友們下面隨著小編來一起學習學習吧2023-06-06
pycharm?使用conda虛擬環(huán)境的詳細配置過程
這篇文章主要介紹了pycharm?使用conda虛擬環(huán)境,本文給大家介紹的非常詳細，對大家的學習或工作具有一定的參考借鑒價值，需要的朋友可以參考下2023-03-03