<深度学习>TensorBoard的demo

import tensorflow.compat.v1 as tf

import os

os.environ["CUDA_VISIBLE_DEVICES"] = "-1"
tf.disable_v2_behavior()
a = tf.constant(4, name = "input_a")
b = tf.constant(2, name = "input_b")
c = tf.multiply(a, b, name = "mul_c")
d = tf.add(a, b, name = "add_d")
e = tf.add(c, d, name = "add_e")

sess = tf.Session()
print(sess.run(e))
write = tf.summary.FileWriter('./my_graph/1', sess.graph)

运行发现当前文件夹下多了个my_graph,里面有个1文件夹,1里面有日志数据

这个文件是用TensorBoard打开的

运行TensorBoard的方法:

进入到保存日志的文件,运行语句:tensorboard --logdir=data (data是保存日志文件的全目录)
成功的话,最下面会显示一个地址,一般是http://localhost:6006/

复制到浏览器打开,就可以看到数据流图啦

猜你喜欢

转载自www.cnblogs.com/painKiller/p/12068796.html
今日推荐