tensorflow模型的保存(三)

训练过程中通常要保存模型

model.save('model.h5')

读取模型

model.load_model('model.h5')

将模型保存为json

model.to_json()

tensorflow模型的保存(三)_第1张图片

保存在本地

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()

你可能感兴趣的:(深度学习,cnn)