快捷導(dǎo)航

淺談pytorch 模型 .pt, .pth, .pkl的區(qū)別及模型保存方式

更新時(shí)間：2020年05月25日 11:41:05 作者：wanghua609

這篇文章主要介紹了淺談pytorch 模型 .pt, .pth, .pkl的區(qū)別及模型保存方式，具有很好的參考價(jià)值，希望對(duì)大家有所幫助。一起跟隨小編過(guò)來(lái)看看吧

我們經(jīng)常會(huì)看到后綴名為.pt, .pth, .pkl的pytorch模型文件，這幾種模型文件在格式上有什么區(qū)別嗎？

其實(shí)它們并不是在格式上有區(qū)別，只是后綴不同而已（僅此而已），在用torch.save()函數(shù)保存模型文件時(shí)，各人有不同的喜好，有些人喜歡用.pt后綴，有些人喜歡用.pth或.pkl.用相同的torch.save（）語(yǔ)句保存出來(lái)的模型文件沒(méi)有什么不同。

在pytorch官方的文檔/代碼里，有用.pt的，也有用.pth的。一般慣例是使用.pth,但是官方文檔里貌似.pt更多，而且官方也不是很在意固定用一種。

模型保存與調(diào)用方式一：

保存：

torch.save(model.state_dict(), mymodel.pth)#只保存模型權(quán)重參數(shù)，不保存模型結(jié)構(gòu)

調(diào)用：

model = My_model(*args, **kwargs) #這里需要重新模型結(jié)構(gòu)，My_model
model.load_state_dict(torch.load(mymodel.pth))#這里根據(jù)模型結(jié)構(gòu)，調(diào)用存儲(chǔ)的模型參數(shù)
model.eval()

模型保存與調(diào)用方式一：

保存：

torch.save(model, mymodel.pth)#保存整個(gè)model的狀態(tài)

調(diào)用：

model=torch.load(mymodel.pth)#這里已經(jīng)不需要重構(gòu)模型結(jié)構(gòu)了，直接load就可以
model.eval()

補(bǔ)充知識(shí)：關(guān)于python 導(dǎo)入包 .pth文件

Python在遍歷已知的庫(kù)文件目錄過(guò)程中，如果見(jiàn)到一個(gè).pth 文件，就會(huì)將文件中所記錄的路徑加入到 sys.path 設(shè)置中，于是 .pth 文件說(shuō)指明的庫(kù)也就可以被 Python 運(yùn)行環(huán)境找到了。

python中有一個(gè).pth文件，該文件的用法是：

首先xxx.pth文件里面會(huì)書(shū)寫(xiě)一些路徑，一行一個(gè)。

將xxx.pth文件放在特定位置，則可以讓python在加載模塊時(shí)，讀取xxx.pth中指定的路徑。

*******************************************

有時(shí)，在用import導(dǎo)入項(xiàng)目文件夾里自己寫(xiě)的python文件時(shí)，常常說(shuō)找不到該文件，這時(shí)可以將項(xiàng)目路徑添加到PYTHONPATH下。

暫時(shí)添加：

import sys
sys.path.append(項(xiàng)目路徑)

這種方法只會(huì)存在內(nèi)存里，一旦退出python就沒(méi)有了

一勞永逸：

在 XXX/pythonXXX/site-packages/ （python 包安裝路徑）下新建后綴為 .pth的文件，在其中加入你的項(xiàng)目路徑，比如

K:\Desktop\face_detect\DFace-win64-master\src
K:\Desktop\face_detect\DFace-win64-master\src\core

*******************************************

有時(shí)候我們正在修改或調(diào)試的程序會(huì)是一個(gè)庫(kù)，為修改方便，我們可能不大希望把它放到 site-packages 下面，而是更愿意把它保留在原始的工程目錄中，以方便 IDE 和版本控制工具進(jìn)行管理。那么怎么能讓 Python 運(yùn)行環(huán)境找到這個(gè)庫(kù)呢？

原理上， Python 運(yùn)行環(huán)境查找?guī)煳募r(shí)本質(zhì)是對(duì) sys.path 列表的遍歷，如果我們想給運(yùn)行環(huán)境注冊(cè)新的類(lèi)庫(kù)進(jìn)來(lái)，

要么得用代碼給 sys.path 列表增加新路徑；

要么得調(diào)整 PYTHONPATH 環(huán)境變量；

要么就得把庫(kù)文件復(fù)制到已經(jīng)在 sys.path 設(shè)置中的路徑中去（比如 site-packages 目錄）；

這些方法都不夠方便。最簡(jiǎn)單的辦法是用 .pth 文件來(lái)實(shí)現(xiàn)。Python 在遍歷已知的庫(kù)文件目錄過(guò)程中，如果見(jiàn)到一個(gè) .pth 文件，就會(huì)將文件中所記錄的路徑加入到 sys.path 設(shè)置中，于是 .pth 文件說(shuō)指明的庫(kù)也就可以被 Python 運(yùn)行環(huán)境找到了。

其實(shí)，easy_install 所依賴(lài)的 egg 包安裝就是靠 site-packages 目錄下的 .pth 文件添加對(duì) egg 包的引用實(shí)現(xiàn)的。所以修改對(duì)應(yīng)的 .pth 文件內(nèi)容，就可以實(shí)現(xiàn)對(duì) egg 包的卸載。

以上這篇淺談pytorch 模型 .pt, .pth, .pkl的區(qū)別及模型保存方式就是小編分享給大家的全部?jī)?nèi)容了，希望能給大家一個(gè)參考，也希望大家多多支持腳本之家。

您可能感興趣的文章: