tensorboard学习笔记

TensorBoard 默认是不会记录每个节点的用时、耗费的内存大小等这些信息的,那么如何才能在图上显示这些信息呢?关键就是如下这些代码,主要就是在 sess.run() 中加入 options 和 run_metadata 参数。

#定义TensorFlow运行选项,设置trace_lever FULL_TRACE
run_options = tf.RunOptions(trace_level=tf.RunOptions.FULL_TRACE)
#定义TensorFlow运行元信息,记录训练运算时间和内存占用等信息
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%03d' % i)
train_writer.add_summary(summary, i)

tensorboard图表解释,https://blog.csdn.net/u010099080/article/details/77426577

posted @ 2018-05-16 15:40  在下小白  阅读(1527)  评论(0编辑  收藏  举报