Saves Checkpoints Every N Steps or Seconds

Saves Checkpoints Every N Steps or Seconds

hook_checkpoint_saver(checkpoint_dir, save_secs = NULL,
  save_steps = NULL, saver = NULL,
  checkpoint_basename = "model.ckpt", scaffold = NULL,
  listeners = NULL)

Arguments

checkpoint_dir

The base directory for the checkpoint files.

save_secs

An integer, indicating saving checkpoints every N secs.

save_steps

An integer, indicating saving checkpoints every N steps.

saver

A saver object, used for saving.

checkpoint_basename

The base name for the checkpoint files.

scaffold

A scaffold, used to get saver object.

listeners

List of checkpoint saver listener subclass instances, used for callbacks that run immediately after the corresponding hook_checkpoint_saver callbacks, only in steps where the hook_checkpoint_saver was triggered.

See also