Tensorflow 笔记 -- tensorboard 的使用
Tensorflow 笔记 -- tensorboard 的使用
TensorFlow提供非常方便的可视化命令Tensorboard,先上代码
import tensorflow as tf
a = tf.constant(5, name="const_a")
b = tf.constant(4, name="const_b")
c = tf.multiply(a, b, name="mul_a_b")
d = tf.add(a, b, name="add_a_b")
e = tf.add(c, d, name="add_c_d")
with tf.Session() as sess:
writer = tf.summary.FileWriter('graph', sess.graph)
print(sess.run(e))
writer.close()
首先是定义一张图,然后跑图。
这里需要注意一点是,FileWriter一定要在sess.run执行前先执行,也就是说,你得把writer = tf.summary.FileWriter('graph',sess.graph)
写在sess.run(e)
之前。参数graph
表示的是你的logdir
目录,后面需要使用。
运行完代码后,直接在代码所在的文件夹目录下执行命令tensorboard --logdir="graph"
。然后会建立一个网站,在浏览器中打开得到的地址即可得到可视化界面。整个过程还是蛮简单的,就是TensorFlow这玩意的API变动的特别大,尤其是步入1.0版。。。较之前发生了比较大的变更。。。希望1.0出来后,API的变更不那么大了吧。。。。。。
夜空中最亮的星,照亮我前行