![]() |
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