升級keras解決load_weights()中的未定義skip_mismatch關鍵字問題
1、問題描述
在用yolov3訓練自己的數(shù)據(jù)集時,嘗試加載預訓練的權重,在凍結前154層的基礎上,利用自己的數(shù)據(jù)集finetune。
出現(xiàn)如下錯誤:
load_weights(),got an unexpected keyword argument skip_mismatch
2、解決方法
因為keras舊版本沒有這一定義,在新的版本中有這一關鍵字的定義,因此,更新keras版本至2.1.5即可解決。
source activate env
pip uninstall keras
pip install keras==2.1.5
此問題解決后,運行訓練代碼,出現(xiàn)如下錯誤:
Loaded runtime CuDNN library:7.0.5 but source was compiled with 7.2.1
這并不是cudnn的問題,將tensorflow的版本從1.11.0降級為1.10.1就好了。
補充知識:TypeError: load() got an unexpected keyword argument 'encoding' 如何解決?
原因是python2和python3的差別。
解決方法把encoding刪掉。
以上這篇升級keras解決load_weights()中的未定義skip_mismatch關鍵字問題就是小編分享給大家的全部內(nèi)容了,希望能給大家一個參考,也希望大家多多支持腳本之家。
相關文章
python PyAUtoGUI庫實現(xiàn)自動化控制鼠標鍵盤
這篇文章主要介紹了python PyAUtoGUI庫實現(xiàn)自動化控制鼠標鍵盤,幫助大家更好的理解和使用python,感興趣的朋友可以了解下2020-09-09python+pyqt5實現(xiàn)KFC點餐收銀系統(tǒng)
這篇文章主要為大家詳細介紹了python+pyqt5實現(xiàn)KFC點餐收銀系統(tǒng),具有一定的參考價值,感興趣的小伙伴們可以參考一下2019-01-01