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