tf.contrib.checkpoint.dot_graph_from_checkpoint
Visualizes an object-based checkpoint (from tf.train.Checkpoint
).
tf.contrib.checkpoint.dot_graph_from_checkpoint( save_path )
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