Tensorflow 2.0下，如何使用tf.summary？答案

【问题标题】：Under Tensorflow 2.0, how to use tf.summary?Tensorflow 2.0下，如何使用tf.summary？
【发布时间】：2019-10-03 06:39:09
【问题描述】：

TF2.0下tf.summary的使用好像和TF1.x不一样。如果我想在 TensorBoard 中可视化多个标量（例如 train_loss、val_loss 等），如何在代码中实现 tf.summary。我想可能是：

writer = tf.summary.create_file_writer('summaries')
for epoch in range(epochs):
    ...
    train_loss = ...
    val_loss = ...
    tf.summary.scalar('train_loss', train_loss)
    tf.summary.scalar('val_loss', val_loss)
    writer.flush()

但它不起作用。我认为问题可能出在 write.flush() 的使用上。

【问题讨论】：

标签： python tensorflow tensorboard tensorflow2.0

【解决方案1】：

在 tensorflow 2.0 中，您需要通过.as_default() 设置默认编写器。此外，step 参数必须在摘要操作中传递。更多详情可以查看here。

writer = tf.summary.create_file_writer("summaries")

for epoch in range(epochs):
    train_loss = ...
    val_loss = ...
    with writer.as_default():
      tf.summary.scalar('train_loss', train_loss, step=epoch)
      tf.summary.scalar('val_loss', val_loss, step=epoch)
    writer.flush()

【讨论】：