TF 2.0 is out! Get hands-on practice at TF World, Oct 28-31. Use code TF20 for 20% off select passes. Register now

tf.train.CheckpointSaverHook

View source on GitHub

Class CheckpointSaverHook

Saves checkpoints every N steps or seconds.

Inherits From: SessionRunHook

Aliases:

  • Class tf.compat.v1.estimator.CheckpointSaverHook
  • Class tf.compat.v1.train.CheckpointSaverHook
  • Class tf.compat.v2.estimator.CheckpointSaverHook
  • Class tf.estimator.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)