当前位置: 代码迷 >> 综合 >> TensorBoard 显示数据流图
  详细解决方案

TensorBoard 显示数据流图

热度:21   发布时间:2023-12-12 16:16:45.0

说明:本文使用 TensorFlow 1.13.0rc2 编写例程。

1. 模型

按照下面的网络建立模型:
在这里插入图片描述

2. TensorFlow 数据流图建模

import tensorflow as tf
a = tf.constant(5, name = "input_a")
b = tf.constant(3, name = "input_b")
c = tf.add(a, b, name = "add_c")
d = tf.multiply(a, b, name = "mul_d")
e = tf.add(c, d, name = "add_e")
sess = tf.Session()
output = sess.run(e)
writer = tf.compat.v1.summary.FileWriter("./my_graph", sess.graph)
writer.close()
sess.close()

3. 生成数据流图文件

我把这段程序保存在 g:\tf_test 目录下,运行后,在g:\tf_test\my_graph 目录下生成一份文件:
在这里插入图片描述

TensorBoard 就是利用这一份文件显示图形的。

4. 启动 TensorBoard 的 Web 服务程序

TensorBoard 利用浏览器显示流程图。
启动 DOS 命令行程序,我用的是 Anaconda3 的命令行窗口:
在这里插入图片描述
在 DOS 命令行下运行:

cd g:\tf_test     //进入前面的程序保存的目录,然后再执行下面的命令
tensorboard --logdir==my_graph: --host=127.0.0.1

截图如下:
在这里插入图片描述
打开浏览器,输入网址:http://127.0.0.1:6006

在这里插入图片描述

注意:我最初用其他浏览器,不能显示图形,换成谷歌浏览器,问题解决。


总结:

  1. 生成图形文件: writer = tf.compat.v1.summary.FileWriter("./my_graph", sess.graph)
  2. 启动 TensorBoard: tensorboard --logdir==my_graph: --host=127.0.0.1
  3. 打开 URL: http://127.0.0.1:6006
  相关解决方案