Tensorboard简单使用

Tensorboard 类似于日志文件,还有画图等功能。主要记录个人学习使用Tensorboard的过程与心得。官网链接

简单使用

本人使用的是环境是 windows10,Anaconda3(python 3.6),Tensorflow-gpu-1.8.0, Tensorboard-1.8.0
1、程序样例

import tensorflow as tf
import numpy as np

batch = 5
z1 = tf.constant([1,0,1,0],dtype=tf.int32)
z2 = tf.constant([1,1,1,1],dtype=tf.int32)
x = tf.cast(z1 >= z2, dtype = tf.float32)
with tf.Session() as sess:
    init = tf.global_variables_initializer()
    sess.run(init)
    writer = tf.summary.FileWriter(r'G:\tf', sess.graph)
    print(sess.run(x))
writer.close()

2、然后打开Anaconda Prompt,输入

tensorboard --logdir="G:\tf"

3、最后在浏览器中输入Anaconda Prompt提示的链接

TensorBoard 1.8.0 at http://zhangxp:6006 (Press CTRL+C to quit)//这是我的链接,因电脑不同而不同

注意事项

本人掉坑里差点没爬出来

  • 因为浏览器域名好像会用到电脑名,因此如果是中文名,第二步会出现解码错误,因此一定要修改为英文名。
    这里写图片描述
  • 第二步中的路径,等号之后不要连空格,直接连路径名,最好用双引号括起来。

其他使用

下面基本描述添加标量(scalars),图(graph),直方图(histogram)方式

tf.summary.scalar('loss', loss)//所有希望显示的变量都可以该方式添加
tf.summary.histogram('histogram', var)//以柱状图显示
merged = tf.summary.merge_all()//汇总数据
with tf.Session(config=config) as sess: 
    train_writer = tf.summary.FileWriter(path, sess.graph)
    sess.run(init)
    for i in range(epoch):
        if(i % 100 == 99)://记录运行内存量,运行时间
            run_options = tf.RunOptions(trace_level=tf.RunOptions.FULL_TRACE)
            run_metadata = tf.RunMetadata()
            summary, _ = sess.run([merged, train_step],
                                  feed_dict=feed_dict(True),
                                  options=run_options,
                                  run_metadata=run_metadata)
            train_writer.add_run_metadata(run_metadata, 'step%d' % i)
            train_writer.add_summary(summary, i)
        else:           
            summary, others = sess.run([merged, other])
            train_writer.add_summary(summary, i)
train_writer.close()

猜你喜欢

转载自blog.csdn.net/hunt_ing/article/details/82048306
今日推荐