训练过程中通常要保存模型
model.save('model.h5')
读取模型
model.load_model('model.h5')
将模型保存为json
model.to_json()
保存在本地
model.to_json()
with open('./training/save.json', 'w') as w:
w.write(model.to_json())
读取
model = keras.models.model_from_json
获取权重参数
model.get_weights()
保存权重
model.save_weights()
加载
model.load_weights()