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)