如何将两个 Keras 模型的摘要数据写入自己的files/folders?

How to write summary data for two Keras models to their own files/folders?

我正在为我的 Tensorflow 模型记录一些自定义指标,如下所示:

# Setup
metric_writer = tf.summary.create_file_writer("{}\train".format(log_dir))
metric_writer.set_as_default()

# During training write the score
tf.summary.scalar("score", score, step=episode)

一切正常,我可以随心所欲地在 Tensorboard 中查看我的数据。但是,我现在想 运行 在我的过程中使用两个模型,并将它们的数据记录为分开 files/folders 以便我可以轻松地比较它们。然而,由于写入数据是用 tf.summary.scalar 完成的,它没有(或接受)对模型或作者的任何引用,我无法弄清楚如何写入两个不同的 files/folders (一个每个型号)。

您可以使用两个单独的 tf.SummaryWriter 及其各自的上下文管理器。除了取自 tensorflow documentation:

summary_writer = tf.summary.create_file_writer('/tmp/summaries')
with summary_writer.as_default():
  tf.summary.scalar('loss', 0.1, step=42)

这样您就可以控制使用哪个摘要编写器。