TensorBoard使用
刚开始使用tensorboard也遇见了一些问题,现在把正确的流程也出来避免大家走一些弯路。
一、编写tensorflow模块写的代码
1. 这段代码是建立计算图与执行
// An highlighted block
# 建立Tensor Flow Variable 变量
import tensorflow as tf
width = tf.placeholder("int32",name="width")
height = tf.placeholder("int32",name="height")
area = tf.multiply(width,height,name="area")
with tf.Session() as sess:
init = tf.global_variables_initializer()
sess.run(init)
print('area=',sess.run(area,feed_dict={width:6,height:8}))
2. 这段代码是将数据写入日志文件,然后由tensorboard读取
# 建立TensorFlow Variable 变量
# 下面的程序代码将要显示在TensorBoard的数据写入log文件。
tf.summary.merge_all() # 将所有要显示在TensorBoard的数据整合
train_writer = tf.summary.FileWriter("log/area",sess.graph) # 将所有要显示在TensorBoard 的数据写入log文件。log文件会存储在当前程序执行目下的lod/area子目录中
3.保存过的路径地址和文件
二、启动tensorboard
windows + R 输入 cmd 启动命令提示符
- 切换到log的保存路径
- 激活tensorflow的虚拟环境
- 启动Tensorboard
logdir 后面是读取的目录
启动成功后结果
三、在TensorBoard查看计算图
启动TensorBoard 之后在浏览器输入网址
http://localhost:6006/
输入网址后,会出现TensorBoard界面,在菜单中选择GRAPHS,之后就会看到计算图如下: