如何将两个 Keras 模型的摘要数据写入它们自己的文件/文件夹

Question

我正在为我的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)

这一切都工作正常，我可以根据需要在张量板中查看我的数据。但是，我现在想在我的过程中运行两个模型，并将它们的数据记录到单独的文件/文件夹中，以便我可以轻松比较它们。但是，由于写入数据是使用没有(或接受(任何对模型或编写器的引用tf.summary.scalar完成的，因此我无法弄清楚如何写入两个不同的文件/文件夹(每个模型一个(。

Answer 1

您可以使用

两个单独的tf.SummaryWriter及其各自的上下文管理器。除了取自张量流文档：

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

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