TensorBoard可以用于显示训练图模型,展示某些量(比如loss, weights)的变化,查看图片等。

TensorBoard读取保存summary数据的文件来获取信息。基本流程为:

  1. 建立graph,并且使用tf.summary.scalar[/histogram/...](name, tensor)指定要获取的tensor
  2. 使用all_summaries = tf.summary.merge_all()将所有summary的Op合并起来
  3. 新建一个summary_writer = tf.summary.FileWriter(logdir, graph=None),指定要保存的数据的路径,还可以指定对应的graph
  4. 一般每隔n次迭代,运行summary = sess.run(all_summaries)获取结果,添加到summary_writer.add_summary(summary, step)中。
  5. 启动tensorboard服务,使用tensorboard --logdir=path/to/log-directorypython -m tensorflow.tensorboard --logdir=...
  6. 打开浏览器,查看localhost:6006获取结果

参考资料:

results matching ""

    No results matching ""