模型的保存和恢复

模型的保存和恢复

1. 模型整体保存

模型整体保存指保存模型权重，模型配置和优化器配置

import tensorflow as tf

# 保存模型
model.save("model.h5")
# 加载模型
new_model = tf.keras.models.load_model("model.h5")

2. 仅保存模型架构

json_config = model.to_json()
# 保存json文件
with open("model_config.json", 'wt') as f:
    f.write