tensorflow模型的保存(三)

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

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

猜你喜欢

转载自blog.csdn.net/monk96/article/details/125768974