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刪掉。
[limiyoyo ] weights()中的未定義skip已經有301次圍觀