国产探花免费观看_亚洲丰满少妇自慰呻吟_97日韩有码在线_资源在线日韩欧美_一区二区精品毛片,辰东完美世界有声小说,欢乐颂第一季,yy玄幻小说排行榜完本

首頁 > 編程 > Python > 正文

pytorch實現從本地加載 .pth 格式模型

2020-02-15 18:24:46
字體:
來源:轉載
供稿:網友

可以從官網加載預訓練好的模型:

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 格式模型就是小編分享給大家的全部內容了,希望能給大家一個參考,也希望大家多多支持武林站長站。

發表評論 共有條評論
用戶名: 密碼:
驗證碼: 匿名發表
主站蜘蛛池模板: 天门市| 东源县| 曲阜市| 崇仁县| 城口县| 翁牛特旗| 康马县| 扶绥县| 灵山县| 敖汉旗| 七台河市| 常山县| 台安县| 芜湖县| 华安县| 穆棱市| 银川市| 麻城市| 铁力市| 共和县| 周至县| 岱山县| 福泉市| 宝山区| 郸城县| 荃湾区| 吉林省| 金寨县| 江达县| 汾西县| 汽车| 同德县| 澄迈县| 枝江市| 噶尔县| 温州市| 梓潼县| 平山县| 阜新市| 环江| 滨州市|