歡迎您光臨本站 註冊首頁

weights()中的未定義skip

←手機掃碼閱讀     limiyoyo @ 2020-06-13 , reply:0

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次圍觀

http://coctec.com/docs/program/show-post-238387.html