在服务器上使用 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')