TensorFlow 2.0 Beta is available Learn more

tf.contrib.tpu.AsyncCheckpointSaverHook

View source on GitHub

Class AsyncCheckpointSaverHook

Saves checkpoints every N steps or seconds.

Inherits From: CheckpointSaverHook

__init__

View source

__init__(
    checkpoint_dir,
    save_secs=None,
    save_steps=None,
    saver=None,
    checkpoint_basename='model.ckpt',
    scaffold=None,
    listeners=None
)

Initializes a CheckpointSaverHook.

Args:

  • checkpoint_dir: str, base directory for the checkpoint files.
  • save_secs: int, save every N secs.
  • save_steps: int, save every N steps.
  • saver: Saver object, used for saving.
  • checkpoint_basename: str, base name for the checkpoint files.
  • scaffold: Scaffold, use to get saver object.
  • listeners: List of CheckpointSaverListener subclass instances. Used for callbacks that run immediately before or after this hook saves the checkpoint.

Raises:

  • ValueError: One of save_steps or save_secs should be set.
  • ValueError: At most one of saver or scaffold should be set.

Methods

after_create_session

View source

after_create_session(
    session,
    coord
)

after_run

View source

after_run(
    run_context,
    run_values
)

before_run

View source

before_run(run_context)

begin

View source

begin()

end

View source

end(session)