簡體   English   中英

如何保存 model 而不是檢查點?

[英]How do I save a model instead of a checkpoint?

我正在關注本教程,但是當我想將 .h5 文件轉換為 .pb 時,我發現我不能使用檢查點。

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

請給我解釋一下

使用 model.save() 將 model 保存為 .pb 格式。 有關更多信息,請閱讀Tensorflow SavedModel 格式

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")

暫無
暫無

聲明:本站的技術帖子網頁,遵循CC BY-SA 4.0協議,如果您需要轉載,請注明本站網址或者原文地址。任何問題請咨詢:yoyou2525@163.com.

 
粵ICP備18138465號  © 2020-2024 STACKOOM.COM