TensorBoard一幅图中显示多条曲线

最新推荐文章于 2026-03-23 05:00:40 发布

转载最新推荐文章于 2026-03-23 05:00:40 发布 · 2.1w 阅读

收录于

TensorFlow 专栏收录该内容

1 篇文章

订阅专栏

本文通过一个具体的示例展示了如何使用TensorFlow框架记录训练过程中的损失值，并将这些数据写入到TensorBoard中进行可视化。示例中创建了两个文件写入器，分别向不同的日志目录写入随机生成的损失值。

Python3.8

Conda

Python

Python 是一种高级、解释型、通用的编程语言，以其简洁易读的语法而闻名，适用于广泛的应用，包括Web开发、数据分析、人工智能和自动化脚本

原文链接点击打开链接

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()