TensorFlow 2.0 Beta is available Learn more

tf.contrib.checkpoint.object_metadata

Retrieves information about the objects in a checkpoint.

tf.contrib.checkpoint.object_metadata(save_path)

Defined in python/training/tracking/util.py.

Example usage:

object_graph = tf.contrib.checkpoint.object_metadata(
    tf.train.latest_checkpoint(checkpoint_directory))
ckpt_variable_names = set()
for node in object_graph.nodes:
  for attribute in node.attributes:
    ckpt_variable_names.add(attribute.full_name)

Args:

Returns:

A parsed tf.contrib.checkpoint.TrackableObjectGraph protocol buffer.

Raises:

  • ValueError: If an object graph was not found in the checkpoint.