TensorBoard一幅图中显示多条曲线

最新推荐文章于 2024-03-30 23:38:14 发布

云生1992

最新推荐文章于 2024-03-30 23:38:14 发布

阅读量2.1w

点赞数 4

分类专栏： TensorFlow

TensorFlow 专栏收录该内容

1 篇文章 0 订阅

订阅专栏

原文链接点击打开链接

import tensorflow as tf
from numpy import random

writer_1 = tf.summary.FileWriter("./logs/plot_1")
writer_2 = tf.summary.FileWriter("./logs/plot_2")

log_var = tf.Variable(0.0)
tf.summary.scalar("loss", log_var)

write_op = tf.summary.merge_all()

session = tf.InteractiveSession()
session.run(tf.global_variables_initializer())

for i in range(100):
    # for writer 1
    summary = session.run(write_op, {log_var: random.rand()})
    writer_1.add_summary(summary, i)
    writer_1.flush()

    # for writer 2
    summary = session.run(write_op, {log_var: random.rand()})
    writer_2.add_summary(summary, i)
    writer_2.flush()