有一些非常流行的網絡如 resnet、squeezenet、densenet等在pytorch里面都有，包括網絡結構和訓練好的模型。

按官網加載預訓練好的模型：

import torchvision.models as models

# pretrained=True就可以使用預訓練的模型

resnet18 = models.resnet18(pretrained=True)

print(resnet18)

報錯如下：

requests.exceptions.ConnectionError: ('Connection aborted.', TimeoutError(10060, '由于連接方在一段時間后沒有正確答復或連接的主機沒有反應，連接嘗試失敗。', None, 10060, None))

主要是因為代碼會去遠端下載模型的參數，而國內的網一般連接不上，這是我們需要手動去下載你要的預訓練網絡。

通過地址下載，地址有兩種獲取方式：

1.從報錯里面獲取，上述代碼運行時會出現這樣一行信息：

Downloading: "https://download.pytorch.org/models/resnet18-5c106cde.pth" to C:\Users\Luo/.torch\models\resnet18-5c106cde.pth

復制這個網址到瀏覽器，有可能打不開，去掉https://，直接輸入download.pytorch.org/models/resnet18-5c106cde.pth就可以下載了。

找到對應模型名稱點進去找地址

總結

以上是生活随笔為你收集整理的python的pth打开方式_pytorch 加载(.pth)格式的模型实例的全部內容，希望文章能夠幫你解決所遇到的問題。

如果覺得生活随笔網站內容還不錯，歡迎將生活随笔推薦給好友。