當前位置: 首頁>>代碼示例 >>用法及示例精選 >>正文

Python tf.train.CheckpointOptions用法及代碼示例


構建檢查點的選項。

用法

tf.train.CheckpointOptions(
    experimental_io_device=None
)

參數

  • experimental_io_device String 。適用於分布式設置。用於訪問文件係統的 TensorFlow 設備。如果None(默認)然後對於每個變量，從分配該變量的主機的 CPU:0 設備訪問文件係統。如果指定，則改為從該設備訪問文件係統以獲取所有變量。

    例如，如果您想保存到本地目錄，例如在分布式設置中運行時的"/tmp"，這很有用。在這種情況下，為可以訪問"/tmp" 目錄的主機傳遞一個設備。

屬性

  • experimental_io_device

用作 tf.train.Checkpoint.save() 或 tf.train.Checkpoint.restore() 方法的 options 參數，以調整變量的保存/恢複方式。

示例：在 "localhost" 上運行 IO 操作，同時保存檢查點：

step = tf.Variable(0, name="step")
checkpoint = tf.train.Checkpoint(step=step)
options = tf.train.CheckpointOptions(experimental_io_device="/job:localhost")
checkpoint.save("/tmp/ckpt", options=options)

相關用法


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