解決pytorch 保存模型遇到的問題
今天用pytorch保存模型時遇到bug
Can't pickle <class 'torch._C._VariableFunctions'>
在google上查找原因，發(fā)現(xiàn)是保存時保存了整個模型的原因，而模型中有一些自定義的參數(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é)果不一樣，這樣會導(dǎo)致訓(xùn)練采用PIL，預(yù)測時采用opencv，結(jié)果差別很大，尤其是在檢測和分割任務(wù)中比較明顯。
3. 數(shù)值計(jì)算
pytorch的torch.exp與c++的exp計(jì)算，10e-6的數(shù)值時候會有10e-3的誤差，對于高精度計(jì)算需要特別注意，比如
兩個輸入5.601597， 5.601601， 經(jīng)過exp計(jì)算后變成270.85862343143174， 270.85970686809225
以上為個人經(jīng)驗(yàn)，希望能給大家一個參考，也希望大家多多支持腳本之家。如有錯誤或未考慮完全的地方，望不吝賜教。
相關(guān)文章
這篇文章主要介紹了深入解析Python中的lambda表達(dá)式的用法,包括其與def之間的區(qū)別,需要的朋友可以參考下2015-08-08
GUI圖形用戶界面編程,我們可以通過python提供的豐富的組件,快速的實(shí)現(xiàn)使用圖形的界面和用戶交互, GUI編程類似于“搭積?”,將?個個組件(Widget)放到窗?中,這篇文章主要給大家介紹了基于Python的GUI圖形用戶界面編程的相關(guān)資料,需要的朋友可以參考下2022-12-12
Python監(jiān)聽鍵盤和鼠標(biāo)事件的示例代碼
這篇文章主要介紹了Python監(jiān)聽鍵盤和鼠標(biāo)事件的示例代碼，幫助大家更好的理解和使用python，提高辦公效率，感興趣的朋友可以了解下2020-11-11
Python實(shí)現(xiàn)matplotlib顯示中文的方法詳解
這篇文章主要介紹了Python實(shí)現(xiàn)matplotlib顯示中文的方法,結(jié)合實(shí)例形式詳細(xì)總結(jié)分析了Python使用matplotlib庫繪圖時顯示中文的相關(guān)操作技巧與注意事項(xiàng),需要的朋友可以參考下2018-02-02
這篇文章主要介紹了Python Thread虛假喚醒概念與防范，虛假喚醒是一種現(xiàn)象，它只會出現(xiàn)在多線程環(huán)境中，指的是在多線程環(huán)境下，多個線程等待在同一個條件上，等到條件滿足時，所有等待的線程都被喚醒，但由于多個線程執(zhí)行的順序不同2023-02-02
Appium+python自動化之連接模擬器并啟動淘寶APP（超詳解）
這篇文章主要介紹了Appium+python自動化之 連接模擬器并啟動淘寶APP（超詳解）本文以淘寶app為例，通過實(shí)例代碼給大家介紹的非常詳細(xì),需要的朋友可以參考下2019-06-06
Python使用cx_Oracle模塊將oracle中數(shù)據(jù)導(dǎo)出到csv文件的方法
這篇文章主要介紹了Python使用cx_Oracle模塊將oracle中數(shù)據(jù)導(dǎo)出到csv文件的方法,涉及Python中cx_Oracle模塊與csv模塊操作Oracle數(shù)據(jù)庫及csv文件的相關(guān)技巧,需要的朋友可以參考下2015-05-05
在大家的日常python程序的編寫過程中，都會有自己解決某個問題的解決辦法，或者是在程序的調(diào)試過程中，用來幫助調(diào)試的程序公式，本文總結(jié)了22個萬用公式，感興趣的可以了解一下2021-07-07
matlab和Excel的數(shù)據(jù)交互操作(非xlsread和xlswrite)
在使用MATLAB時,可能會遇到很多表格數(shù)據(jù)的處理,有時MATLAB也需要利用現(xiàn)存的表格數(shù)據(jù)實(shí)現(xiàn)操作目的,下面這篇文章主要給大家介紹了關(guān)于matlab和Excel的交互操作的相關(guān)資料,非xlsread和xlswrite,需要的朋友可以參考下2021-08-08
本文主要給大家講解如何使用pyqt5打開攝像頭并在label上顯示，以及優(yōu)化成圓形label框。通過實(shí)例代碼給大家介紹的非常詳細(xì)，感興趣的朋友跟隨小編一起看看吧2022-01-01