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