时间戳

首页 > 解决方案 > 如何在 TF2.0 的张量板中将标量放在同一行？

tensorboard - 如何在 TF2.0 的张量板中将标量放在同一行？

问题描述

如果我在张量板上有 3 个不同的标量，并且我想将其中两个放在同一行中，我该怎么办？以下代码生成具有 3 个不同行的张量板。我想把前两个放在一排，第二个放在另一排。

with summary_writer.as_default():
    tf.summary.scalar("training_loss", loss, step=batch_index)
    tf.summary.scalar("learning_rate", optimizer.learning_rate, step=batch_index)
with summary_writer.as_default():
    tf.summary.scalar("validation_loss", val_loss, step=batch_index)

标签： tensorboardtensorflow2.0

解决方案

在 TensorFlow GitHub 上提出这个问题后，我得到了答案。链接是https://github.com/tensorflow/tensorboard/issues/2984。

基本上，只需在标量名称之前添加类别名称：

with summary_writer.as_default():
    tf.summary.scalar("training/training_loss", loss, step=batch_index)
    tf.summary.scalar("training/learning_rate", optimizer.learning_rate, step=batch_index)
with summary_writer.as_default():
    tf.summary.scalar("validation/validation_loss", val_loss, step=batch_index)

推荐阅读