溫馨提示×
您好，登錄后才能下訂單哦！
點擊 登錄注冊 即表示同意《億速云用戶服務條款》
您好，登錄后才能下訂單哦！
這篇文章將為大家詳細講解有關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) 為對應模型的函數即可。
1.PyTorch是相當簡潔且高效快速的框架；2.設計追求最少的封裝；3.設計符合人類思維，它讓用戶盡可能地專注于實現自己的想法；4.與google的Tensorflow類似，FAIR的支持足以確保PyTorch獲得持續的開發更新；5.PyTorch作者親自維護的論壇 供用戶交流和求教問題6.入門簡單
關于“pytorch怎么實現從本地加載 .pth格式模型”這篇文章就分享到這里了，希望以上內容可以對大家有一定的幫助，使各位可以學到更多知識，如果覺得文章不錯，請把它分享出去讓更多的人看到。
免責聲明：本站發布的內容（圖片、視頻和文字）以原創、轉載和分享為主，文章觀點不代表本網站立場，如果涉及侵權請聯系站長郵箱：is@yisu.com進行舉報，并提供相關證據，一經查實，將立刻刪除涉嫌侵權內容。
