tf.contrib.checkpoint.dot_graph_from_checkpoint

Визуализирует чекпоинт на основе объектов (из tf.train.Checkpoint).

tf.contrib.checkpoint.dot_graph_from_checkpoint(
    save_path
)

Полезно для проверки чекпоинтов и отладки проблем с загрузкой.

Пример использования из Python (требуется 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')

Пример использования в командной строке:

python -c "import tensorflow as tf;\
  print(tf.contrib.checkpoint.dot_graph_from_checkpoint('/path/to/ckpt'))"\
  | dot -Tsvg > /tmp/tensorflow/checkpoint_viz.svg