Keras 建立断点checkpoint;保存,加载模型

参考链接:

如何为Keras中的深度学习模型建立Checkpoint

Keras框架训练模型保存及再载入

模型保存与加载:

保存:

#保存模型

model.save('model.h5')

# 保存参数

model.save_weights('my_model_weights.h5')

加载:

from keras.models import load_model

#载入模型、参数

model = load_model('model.h5')

model.load_weights('my_model_weights.h5')

# 保存网络结构,载入网络结构
from keras.models import model_from_json
json_string = model.to_json()
model = model_from_json(json_string)

print(json_string)
 

发布了10 篇原创文章 · 获赞 10 · 访问量 7517

猜你喜欢

转载自blog.csdn.net/qq_41647438/article/details/101099437
今日推荐