如何保存 model 而不是检查点？

Question

I was following this tutorial, but when I wanted to convert the.h5 file into.pb I found out that I can't use checkpoints.

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

Please explain this to me

Answer 1

Use model.save() to save the model in.pb format. For more information read 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")