tf.train.list_variables

Перечисляет ключи контрольных точек и формы переменных в контрольной точке.

tf.train.list_variables(
    ckpt_dir_or_file
)

Ключи контрольных точек — это пути в графе контрольной точки.

Пример использования:

import tensorflow as tf
import os
ckpt_directory = "/tmp/training_checkpoints/ckpt"
ckpt = tf.train.Checkpoint(optimizer=optimizer, model=model)
manager = tf.train.CheckpointManager(ckpt, ckpt_directory, max_to_keep=3)
train_and_checkpoint(model, manager)
tf.train.list_variables(manager.latest_checkpoint)