TensorFlow 2.0 Beta is available Learn more

tf.contrib.training.wait_for_new_checkpoint

View source on GitHub

Waits until a new checkpoint file is found.

tf.contrib.training.wait_for_new_checkpoint(
    checkpoint_dir,
    last_checkpoint=None,
    seconds_to_sleep=1,
    timeout=None
)

Args:

  • checkpoint_dir: The directory in which checkpoints are saved.
  • last_checkpoint: The last checkpoint path used or None if we're expecting a checkpoint for the first time.
  • seconds_to_sleep: The number of seconds to sleep for before looking for a new checkpoint.
  • timeout: The maximum number of seconds to wait. If left as None, then the process will wait indefinitely.

Returns:

a new checkpoint path, or None if the timeout was reached.