Python tf.train.Checkpoint.write用法及代碼示例

用法

write(
    file_prefix, options=None
)

編寫一個訓練檢查點。

檢查點包括由該對象創建的變量以及在調用Checkpoint.write() 時它所依賴的任何可跟蹤對象。

write 不對檢查點編號、增加 save_counter 或更新 tf.train.latest_checkpoint 使用的元數據。它主要供更高級別的檢查點管理實用程序使用。 save 提供了這些函數的非常基本的實現。

使用 write 編寫的檢查點必須使用 read 讀取。

示例用法：

step = tf.Variable(0, name="step")
checkpoint = tf.Checkpoint(step=step)
checkpoint.write("/tmp/ckpt")

# Later, read the checkpoint with read()
checkpoint.read("/tmp/ckpt")

# You can also pass options to write() and read(). For example this
# runs the IO ops on the localhost:
options = tf.CheckpointOptions(experimental_io_device="/job:localhost")
checkpoint.write("/tmp/ckpt", options=options)

# Later, read the checkpoint with read()
checkpoint.read("/tmp/ckpt", options=options)