TF 2.0 is out! Get hands-on practice at TF World, Oct 28-31. Use code TF20 for 20% off select passes. Register now

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