Missed TensorFlow World? Check out the recap. Learn more

tf.contrib.model_pruning.graph_def_from_checkpoint

View source on GitHub

Converts checkpoint data to GraphDef.

tf.contrib.model_pruning.graph_def_from_checkpoint(
    checkpoint_dir,
    output_node_names
)

Reads the latest checkpoint data and produces a GraphDef in which the variables have been converted to constants.

Args:

  • checkpoint_dir: Path to the checkpoints.
  • output_node_names: List of name strings for the result nodes of the graph.

Returns:

A GraphDef from the latest checkpoint

Raises:

  • ValueError: if no checkpoint is found