tf.contrib.checkpoint.dot_graph_from_checkpoint

Visualizes an object-based checkpoint (from tf.train.Checkpoint).

Useful for inspecting checkpoints and debugging loading issues.

Example usage from Python (requires pydot):

import tensorflow as tf
import pydot

dot_string = tf.contrib.checkpoint.dot_graph_from_checkpoint('/path/to/ckpt')
parsed, = pydot.graph_from_dot_data(dot_string)
parsed.write_svg('/tmp/tensorflow/visualized_checkpoint.svg')

Example command line usage:

python -c "import tensorflow as tf;\
  print(tf.contrib.checkpoint.dot_graph_from_checkpoint('/path/to/ckpt'))"\
  | dot -Tsvg > /tmp/tensorflow/checkpoint_viz.svg
Args
save_path The checkpoint prefix, as returned by tf.train.Checkpoint.save or tf.train.latest_checkpoint.
Returns
A graph in DOT format as a string.

© 2020 The TensorFlow Authors. All rights reserved.
Licensed under the Creative Commons Attribution License 3.0.
Code samples licensed under the Apache 2.0 License.
https://www.tensorflow.org/versions/r1.15/api_docs/python/tf/contrib/checkpoint/dot_graph_from_checkpoint