在TF2.0中,如何将标量放在张量板的同一行中?

时间:2019-11-03 12:50:37

标签: tensorboard tensorflow2.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)

1 个答案:

答案 0 :(得分:1)

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