首页 > 解决方案 > Tensorflow:如何在 tensorboard 中查看检查点?

问题描述

假设我有内容检查点:

checkpoint
├── model.ckpt-240000.data-00000-of-00001
├── model.ckpt-240000.index
└── model.ckpt-240000.meta

是否可以在 tensorboard 中查看检查点的内容,或者只能转换为.pb格式?

标签: pythontensorflowdeep-learningtensorboard

解决方案


看起来可以这样做:

import tensorflow as tf

g = tf.Graph()

with g.as_default() as g:
    tf.train.import_meta_graph('./checkpoint/model.ckpt-240000.meta')

with tf.Session(graph=g) as sess:
    file_writer = tf.summary.FileWriter(logdir='checkpoint_log_dir/faceboxes', graph=g)

进而tensorboard --logdir checkpoint_log_dir/faceboxes/


推荐阅读