1. 程式人生 > 程式設計 >升級keras解決load_weights()中的未定義skip_mismatch關鍵字問題

升級keras解決load_weights()中的未定義skip_mismatch關鍵字問題

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關鍵字問題就是小編分享給大家的全部內容了,希望能給大家一個參考,也希望大家多多支援我們。