tensorflow中Tensorboard的用法

tensorflow中Tensorboard的用法

下面代码定义了一个简单的用于实现常量加法和乘法的计算图。

import tensorflow as tf
input1 = tf.constant(3.0, # 常量的值
                     name='A' # 定义名称，会在tensorboard中代替显示
                     )
input2 = tf.constant(4.0,name='B')
input3 = tf.constant(5.0,name='C')
#加法
add = tf.add(input2,input3)
#乘法
mul = tf.multiply(input1,add)
# 创建writer，写日志文件
# 'logs'：文件保存的路径
wirter = tf.summary.FileWriter('logs',tf.get_default_graph())
writer.close() 

在上面程序的12、13行中，创建一个writer，将tensorboard summary写入文件夹/logs，然后运行上面的程序，在程序定义的日志文件夹/path/to/logs目录下，生成了一个新的日志文件events.out.tfevents.1524711020.bdi-172，如下图1所示。当然，这里的日志文件夹也可以由读者自行指定，但是要确保文件夹存在。如果使用的tensorboard版本比较低，那么直接运行上面的代码可能会报错，此时，可以尝试将第13行代码改为file_writer=tf.train.SummaryWriter(‘logs’, sess.graph)

在终端下执行命令：

tensorboard --logdir=文件保存路径 --host=127.0.0.1

或者写个tensor_board.py文件，运行：

logdir = './logs' #文件保存路径
import os
os.system("tensorboard --logdir=%s --host=127.0.0.1" % (logdir))

出现下图：

在浏览器上输入：

http://127.0.0.1:6006/

浏览器上显示：