TensorBoard可以用于显示训练图模型,展示某些量(比如loss, weights)的变化,查看图片等。
TensorBoard读取保存summary数据的文件来获取信息。基本流程为:
- 建立graph,并且使用
tf.summary.scalar[/histogram/...](name, tensor)指定要获取的tensor - 使用
all_summaries = tf.summary.merge_all()将所有summary的Op合并起来 - 新建一个
summary_writer = tf.summary.FileWriter(logdir, graph=None),指定要保存的数据的路径,还可以指定对应的graph - 一般每隔n次迭代,运行
summary = sess.run(all_summaries)获取结果,添加到summary_writer.add_summary(summary, step)中。 - 启动tensorboard服务,使用
tensorboard --logdir=path/to/log-directory或python -m tensorflow.tensorboard --logdir=... - 打开浏览器,查看
localhost:6006获取结果
参考资料: