如何保存模型而不是检查点？

【问题标题】：How do I save a model instead of a checkpoint?如何保存模型而不是检查点？
【发布时间】：2022-01-19 16:06:35
【问题描述】：

我一直在学习本教程，但是当我想将 .h5 文件转换为 .pb 时，我发现我无法使用检查点。

https://www.thepythoncode.com/article/skin-cancer-detection-using-tensorflow-in-python

请给我解释一下

【问题讨论】：

请提供足够的代码，以便其他人更好地理解或重现问题。

标签： python tensorflow

【解决方案1】：

使用 model.save() 将模型保存为 .pb 格式。更多信息请阅读Tensorflow SavedModel format。

import tensorflow as tf
#Load .h5 saved model
pre_model = tf.keras.models.load_model("final_model.h5")
#Converts .pb format
pre_model.save("saved_model/my_model")

【讨论】：