pytorch怎么實(shí)現(xiàn)從本地加載 .pth格式模型

這篇文章將為大家詳細(xì)講解有關(guān)pytorch怎么實(shí)現(xiàn)從本地加載 .pth格式模型，小編覺得挺實(shí)用的，因此分享給大家做個參考，希望大家閱讀完這篇文章后可以有所收獲。

可以從官網(wǎng)加載預(yù)訓(xùn)練好的模型：

import torchvision.models as models
 
model = models.vgg16(pretrained = True)
print(model)

但是經(jīng)常會出現(xiàn)因?yàn)橄螺d速度太慢而出現(xiàn)requests.exceptions.ConnectionError: ('Connection aborted.', TimeoutError(10060, '由于連接方在一段時間后沒有正確答復(fù)或連接的主機(jī)沒有反應(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的優(yōu)點(diǎn)

1.PyTorch是相當(dāng)簡潔且高效快速的框架；2.設(shè)計追求最少的封裝；3.設(shè)計符合人類思維，它讓用戶盡可能地專注于實(shí)現(xiàn)自己的想法；4.與google的Tensorflow類似，F(xiàn)AIR的支持足以確保PyTorch獲得持續(xù)的開發(fā)更新；5.PyTorch作者親自維護(hù)的論壇 供用戶交流和求教問題6.入門簡單

關(guān)于“pytorch怎么實(shí)現(xiàn)從本地加載 .pth格式模型”這篇文章就分享到這里了，希望以上內(nèi)容可以對大家有一定的幫助，使各位可以學(xué)到更多知識，如果覺得文章不錯，請把它分享出去讓更多的人看到。