解決pytorch 保存模型遇到的問題
今天用pytorch保存模型時(shí)遇到bug
Can't pickle <class 'torch._C._VariableFunctions'>
在google上查找原因，發(fā)現(xiàn)是保存時(shí)保存了整個(gè)模型的原因，而模型中有一些自定義的參數(shù)
將 torch.save(model,save_path) 改為 torch.save(model.state_dict(),save_path)
然后載入模型也做相應(yīng)的更改就好了
補(bǔ)充：pytorch訓(xùn)練模型的一些坑
1. 圖像讀取
opencv的python和c++讀取的圖像結(jié)果不一致，是因?yàn)閜ython和c++采用的opencv版本不一樣，從而使用的解碼庫不同，導(dǎo)致讀取的結(jié)果不同。
2. 圖像變換
PIL和pytorch的圖像resize操作，與opencv的resize結(jié)果不一樣，這樣會(huì)導(dǎo)致訓(xùn)練采用PIL，預(yù)測(cè)時(shí)采用opencv，結(jié)果差別很大，尤其是在檢測(cè)和分割任務(wù)中比較明顯。
3. 數(shù)值計(jì)算
pytorch的torch.exp與c++的exp計(jì)算，10e-6的數(shù)值時(shí)候會(huì)有10e-3的誤差，對(duì)于高精度計(jì)算需要特別注意，比如
兩個(gè)輸入5.601597， 5.601601， 經(jīng)過exp計(jì)算后變成270.85862343143174， 270.85970686809225
以上為個(gè)人經(jīng)驗(yàn)，希望能給大家一個(gè)參考，也希望大家多多支持腳本之家。如有錯(cuò)誤或未考慮完全的地方，望不吝賜教。
相關(guān)文章
這篇文章主要介紹了Python實(shí)現(xiàn)FM算法解析，文中通過示例代碼介紹的非常詳細(xì)，對(duì)大家的學(xué)習(xí)或者工作具有一定的參考學(xué)習(xí)價(jià)值，需要的朋友們下面隨著小編來一起學(xué)習(xí)學(xué)習(xí)吧2019-06-06
這篇文章主要介紹了python爬蟲 Pyppeteer使用方法解析,文中通過示例代碼介紹的非常詳細(xì)，對(duì)大家的學(xué)習(xí)或者工作具有一定的參考學(xué)習(xí)價(jià)值,需要的朋友可以參考下2019-09-09
今天小編就為大家分享一篇解決python 找不到module的問題，具有很好的參考價(jià)值，希望對(duì)大家有所幫助。一起跟隨小編過來看看吧2020-02-02
這篇文章主要介紹了關(guān)于NumPy中asarray的用法及說明，具有很好的參考價(jià)值，希望對(duì)大家有所幫助。如有錯(cuò)誤或未考慮完全的地方，望不吝賜教2022-07-07
Pytorch中的數(shù)據(jù)轉(zhuǎn)換Transforms與DataLoader方式
這篇文章主要介紹了Pytorch中的數(shù)據(jù)轉(zhuǎn)換Transforms與DataLoader方式，具有很好的參考價(jià)值，希望對(duì)大家有所幫助。如有錯(cuò)誤或未考慮完全的地方，望不吝賜教2023-02-02
使用Python解決常見格式圖像讀取nii,dicom,mhd
這篇文章主要介紹了使用Python解決常見格式圖像讀取nii,dicom,mhd,下文具體操作過程需要的小伙伴可以參考一下2022-04-04
淺談keras中的Merge層(實(shí)現(xiàn)層的相加、相減、相乘實(shí)例)
這篇文章主要介紹了淺談keras中的Merge層(實(shí)現(xiàn)層的相加、相減、相乘實(shí)例)，具有很好的參考價(jià)值，希望對(duì)大家有所幫助。一起跟隨小編過來看看吧2020-05-05
Django生成PDF文檔顯示網(wǎng)頁上以及PDF中文顯示亂碼的解決方法
今天小編就為大家分享一篇Django生成PDF文檔顯示網(wǎng)頁上以及PDF中文顯示亂碼的解決方法，具有很好的參考價(jià)值，希望對(duì)大家有所幫助。一起跟隨小編過來看看吧2019-12-12
這篇文章主要給大家介紹了關(guān)于django限制匿名用戶訪問及重定向的相關(guān)資料，文中通過示例代碼介紹的非常詳細(xì)，對(duì)大家的學(xué)習(xí)或者工作具有一定的參考學(xué)習(xí)價(jià)值，需要的朋友們下面隨著小編來一起學(xué)習(xí)學(xué)習(xí)吧。2018-02-02
這篇文章主要給大家介紹了關(guān)于Python排序函數(shù)使用的相關(guān)資料，文中通過示例代碼介紹的非常詳細(xì)，對(duì)大家的學(xué)習(xí)或者工作具有一定的參考學(xué)習(xí)價(jià)值，需要的朋友們下面隨著小編來一起學(xué)習(xí)學(xué)習(xí)吧2020-12-12