Python tf.keras.Model.save用法及代碼示例

用法

save(
    filepath, overwrite=True, include_optimizer=True, save_format=None,
    signatures=None, options=None, save_traces=True
)

參數

filepath 字符串、PathLike、SavedModel 或 H5 文件的路徑以保存模型。
overwrite 是靜默覆蓋目標位置的任何現有文件，還是為用戶提供手動提示。
include_optimizer 如果為 True，則將優化器的狀態一起保存。
save_format 'tf' 或 'h5' ，表示是將模型保存到 Tensorflow SavedModel 還是 HDF5。在 TF 2.X 中默認為 'tf'，在 TF 1.X 中默認為 'h5'。
signatures 使用 SavedModel 保存的簽名。僅適用於'tf' 格式。有關詳細信息，請參閱tf.saved_model.save 中的signatures 參數。
options (僅適用於 SavedModel 格式)tf.saved_model.SaveOptions 對象，指定保存到 SavedModel 的選項。
save_traces (僅適用於 SavedModel 格式)啟用後，SavedModel 將存儲每一層的函數軌跡。這可以禁用，以便隻存儲每一層的配置。默認為 True 。禁用此函數將減少序列化時間並減小文件大小，但它要求所有自定義層/模型都實現get_config() 方法。

將模型保存到 Tensorflow SavedModel 或單個 HDF5 文件。

有關詳細信息，請參閱tf.keras.models.save_model 或序列化和保存指南。

例子：

from keras.models import load_model

model.save('my_model.h5')  # creates a HDF5 file 'my_model.h5'
del model  # deletes the existing model

# returns a compiled model
# identical to the previous one
model = load_model('my_model.h5')

相關用法

注：本文由純淨天空篩選整理自tensorflow.org大神的英文原創作品 tf.keras.Model.save。非經特殊聲明，原始代碼版權歸原作者所有，本譯文未經允許或授權，請勿轉載或複製。