Python tf.train.Checkpoint.read用法及代码示例

用法

read(
    save_path, options=None
)

读取用 write 编写的训练检查点。

读取此 Checkpoint 及其依赖的任何对象。

此方法与restore() 类似，但不期望检查点中的save_counter 变量。它只恢复检查点已经依赖的对象。

该方法主要供使用write() 而不是save() 并具有自己的机制来编号和跟踪检查点的更高级别的检查点管理实用程序使用。

示例用法：

# Create a checkpoint with write()
ckpt = tf.train.Checkpoint(v=tf.Variable(1.))
path = ckpt.write('/tmp/my_checkpoint')

# Later, load the checkpoint with read()
# With restore() assert_consumed() would have failed.
checkpoint.read(path).assert_consumed()

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