升級keras解決load_weights()中的未定義skip_mismatch關鍵字問題
阿新 • • 發佈:2020-06-14
1、問題描述
在用yolov3訓練自己的資料集時,嘗試載入預訓練的權重,在凍結前154層的基礎上,利用自己的資料集finetune。
出現如下錯誤:
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
此問題解決後,執行訓練程式碼,出現如下錯誤:
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關鍵字問題就是小編分享給大家的全部內容了,希望能給大家一個參考,也希望大家多多支援我們。