View source on GitHub
|
Retrieves information about the objects in a checkpoint.
tf.contrib.checkpoint.object_metadata(save_path)
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:
save_path: The path to the checkpoint, as returned bysaveortf.train.latest_checkpoint.
Returns:
A parsed tf.contrib.checkpoint.TrackableObjectGraph protocol buffer.
Raises:
ValueError: If an object graph was not found in the checkpoint.