TensorFlow 2 version | View source on GitHub |
Returns list of all variables in the checkpoint.
tf.train.list_variables(
ckpt_dir_or_file
)
Args | |
---|---|
ckpt_dir_or_file
|
Directory with checkpoints file or path to checkpoint. |
Returns | |
---|---|
List of tuples (name, shape) .
|