您可以通过以下两种方法之一在 TensorFlow 中保存和恢复模型和变量:
tf.train.Saver
tf.saved_model_builder.SavedModelBuilder
SavedModel
让我们看看两种方法的实际应用。
您可以按照 Jupyter 笔记本中的代码ch-11a_Saving_and_Restoring_TF_Models。
ch-11a_Saving_and_Restoring_TF_Models