将多个标量保存到TensorBoard

时间:2019-05-18 01:43:01

标签: python tensorflow tensorboard eager-execution

我处于“急切模式”,并且试图绘制TensorBoard中某些标量的演变。 我已经设法通过使用以下方法来做到这一点-损失函数-

        summary_writer = tf.contrib.summary.create_file_writer(log_dir, flush_millis=10000)
        with summary_writer.as_default(), tf.contrib.summary.always_record_summaries():
        tf.contrib.summary.scalar("loss", curr_loss)

但是,如果我添加另一行代码

        tf.contrib.summary.scalar("phi", phi)

那么它就不会被记录(或者至少它不会在TensorBoard中显示)。我找不到很多参考,但是我找到的参考让我这样写...

有人知道如何正确执行此操作吗?谢谢!

1 个答案:

答案 0 :(得分:1)

如果您紧跟guide,将需要3个关键步骤来编写摘要: 1.创建摘要

phi_summary_op = tf.contrib.summary.scalar("phi", phi)

1.2。 (可选)一次收集所有摘要

merged = tf.summary.merge_all()

2。在会话中执行摘要操作以及训练(或验证)步骤:

summary, _ = sess.run([merged, train_step], feed_dict=feed_dict(True))

3。将结果写入相应的日志文件:

train_writer = tf.summary.FileWriter(FLAGS.summaries_dir + '/train',
                                      sess.graph)
train_writer.add_summary(summary, i)

我希望对您有帮助