【598】解決 Keras 保存模型無法加載使用的問題

參考:Keras模型保存與加載的問題


 

  【問題描述】

  正常來說,model.save("model.h5"),然後通過 load_model("model.h5") 就可以使用了,但是在具體操作的過程中報錯,有博客說明只有 Sequential 構建的模型可以通過這個方法直接加載,目前沒有證實。

  【解決方案】

  按照原始構建模型的方式構建,只是 pass 掉 model.fit() 這個部分,也就是構建跟原始模型結構一致的模型,然後通過加載權重的方式來加載模型

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

發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章