如何在TenworFlow2中使用tf.gradienttape保存和加载模型

Question

我正在使用tf.gradienttape进行模型训练，并且成功保存了每个纪元的检查点。

const makeStyle = (styleCallback: (theme: Theme, props: T) => any) => {
    return <T>(options: T) =>{
        theme = getTheme();
        return styleCallback(theme, options)
    }
}

以下两个问题， 1.如何保存此模型？ 2.以后如何加载该模型？

我的模型是一种自动编码器类型的模型，因此有必要创建重构的模型以比较并查看错误。

Answer 1

本指南可能会有所帮助： https://www.tensorflow.org/guide/saved_model

tf.saved_model.save(model, "Path")

Answer 2

使用load_model API保存和加载模型。

model.save(model_path) 

和

loaded = tf.keras.models.load_model(model_path)

check this tensorflow tutorial