tf.compat.v1.train.generate_checkpoint_state_proto

View source on GitHub

Generates a checkpoint state proto.

tf.compat.v1.train.generate_checkpoint_state_proto(
    save_dir, model_checkpoint_path, all_model_checkpoint_paths=None,
    all_model_checkpoint_timestamps=None, last_preserved_timestamp=None
)

save_dir Directory where the model was saved.
model_checkpoint_path The checkpoint file.
all_model_checkpoint_paths List of strings. Paths to all not-yet-deleted checkpoints, sorted from oldest to newest. If this is a non-empty list, the last element must be equal to model_checkpoint_path. These paths are also saved in the CheckpointState proto.
all_model_checkpoint_timestamps A list of floats, indicating the number of seconds since the Epoch when each checkpoint was generated.
last_preserved_timestamp A float, indicating the number of seconds since the Epoch when the last preserved checkpoint was written, e.g. due to a keep_checkpoint_every_n_hours parameter (see tf.contrib.checkpoint.CheckpointManager for an implementation).

CheckpointState proto with model_checkpoint_path and all_model_checkpoint_paths updated to either absolute paths or relative paths to the current save_dir.

ValueError If all_model_checkpoint_timestamps was provided but its length does not match all_model_checkpoint_paths.