使用tensorboard调试模型

概述

截至目前, tensorboard还是神经网络开发调试中比较好的工具, 各大框架厂商都有直接支持或间接借鉴.

以下描述使用方法.

1利用summary记录要显示的信息

1.1 添加要记录的各种类型的变量

scalar/histogram......

tf.summary.histogram('logits', logits)
tf.summary.scalar('loss', loss)
......

1.2打包变量名

merged = tf.summary.merge_all()

 

1.3添加内存和时间要素

# Define metadata variable and RunOptions
run_options = tf.RunOptions(trace_level=tf.RunOptions.FULL_TRACE)
run_metadata = tf.RunMetadata()

 

1.4 打开一个记录器

Open a writer to write summaries.

writer = tf.summary.FileWriter(FLAGS.log_dir, session.graph)

1.5训练