keras load_model ValueError: Unknown initializer: GlorotUniform


在服務器上使用 keras 對數據進行訓練完成后,通過 save 的方法將模型保存成對應的 xxxxx.h5 文件,將對應的 h5 文件拷貝到 windows 下面的時候通過 load 命令加載卻報錯。

報錯如下:

keras load_model ValueError: Unknown initializer: GlorotUniform。

這是因為在服務器上安裝 keras 時候是在 tensorflow 里面進行的,而 Windows 下面進行加載的時候導入的卻是 keras 下面的加載模型,把對應的導入模塊進行替換即可。

from keras.models import load_model
# 將上面的這句話替換成下面的,即可。
from tensorflow.keras.models import load_model

# 在執行此句就不會報錯了
model = load_model('lstm_model.h5')

 


免責聲明!

本站轉載的文章為個人學習借鑒使用,本站對版權不負任何法律責任。如果侵犯了您的隱私權益,請聯系本站郵箱yoyou2525@163.com刪除。



 
粵ICP備18138465號   © 2018-2025 CODEPRJ.COM