【598】解决 Keras 保存模型无法加载使用的问题

参考:Keras模型保存与加载的问题


  【问题描述】

  正常来说,model.save("model.h5"),然后通过 load_model("model.h5") 就可以使用了,但是在具体操作的过程中报错,有博客说明只有 Sequential 构建的模型可以通过这个方法直接加载,目前没有证实。

  【解决方案】

  按照原始构建模型的方式构建,只是 pass 掉 model.fit() 这个部分,也就是构建跟原始模型结构一致的模型,然后通过加载权重的方式来加载模型

  也就是 model.load_weights("model.h5"),然后再使用的时候是可以成功的。

原文地址:https://www.cnblogs.com/alex-bn-lee/p/14977041.html