假设想要输出loss随epoch的变化图:
1、构建图过程中声明scalar变量
在图中定义loss节点后,加一个
loss_summary = tf.summry.scalar('loss',loss)
当有多个要输出时,可以用
merged_summary_op = tf.merge_all_summaries()
汇总到一个节点,执行时执行这个节点,add这个节点就行
2、构建完整个图后加File_Writer
构建完图,在执行前,声明
file_writer = tf.summary.FileWriter(logdir,tf.get_default_graph())
比如在全局变量init语句之后
3 、执行过程中
喂入数据后run上面的loss_summary节点
然后:
file_writer.add_summary(loss_summary_str,epoch)