tensorboard的简单应用
import tensorflow as tf
tf.reset_default_graph()
logdir="C:/tensorflow_log"
value =tf.Variable( 0, name= "value" )
value1 =tf.Variable( 0, name= "value" )
one =tf.constant( 1 )
new_value=tf.add (value, one )
new_value1=tf.add (value1, value)
update_value= tf.assign(value,new_value)
update_value1= tf.assign(value1,new_value1)
init= tf.global_variables_initializer()
with tf.Session() as sess:
sess.run(init)
for i in range(10):
sess.run(update_value)
print(sess.run(value))
sess.run(update_value1)
print(sess.run(value1))
writer=tf.summary.FileWriter(logdir,tf.get_default_graph())
writer.close()
其实就是四句关键的代码:
tf.reset_default_graph() # 清除默认计算图,防止干扰
logdir="C:/tensorflow_log" # 设置log保存的位置
writer=tf.summary.FileWriter(logdir,tf.get_default_graph()) #生成一个写日志的writer,把当前的计算图写入指定位置
writer.close()
接着就是如何打开tensorboard,打开cmd,进入你存放计算图日志的路径
接着输入 tensorboard --logdir=xxx(你存放日志的路径)
然后复制tensorboard给出的服务地址,进入浏览器,粘贴,回车,搞定。
此外,启动服务的端口默认为6006,可以使用 --port 参数 去修改服务的端口。