tensorboard是tensorflow的可视化工具。主要是对变量和网络结构的显示,变量分为两种类型scalar和histogram,其中scalar主要保存scalar型变量,如accuracy, loss。histogram保存高维数据的分布,如权重等。具体操作流程为:
这里使用的保存工具是tf.summary.FileWriter(path, graph)
, 如果指定了graph那么就会把图一起起保存进去。
如果既可视化训练过程,也要可视化测试过程,就要有两个文件分别保存op运行的数据,最后tensorboard --logdir
时使用的路径是两个子文件夹的上一层目录,这样tensorboard会分别显示子目录下的保存信息。
具体代码参考官网的例程。
https://www.tensorflow.org/programmers_guide/summaries_and_tensorboard