标量摘要的张量板平均值

Question

我正在尝试使用 TensorBoard 可视化我的输出损失值。我这样做的方式很简单。我只是在小批量的每次迭代之后添加一个 summary.scalar 。在一个时代结束时，张量板将向我展示每次迭代后的所有损失。我想获得所有这些迭代的平均值，这意味着我希望 tensorboard 向我显示每个时期的一个损失值。

谢谢。

Answer 1

尝试创建另一个标量平均值，并为一个时期中的每一步计算一个运行平均值，如下所示：

average = 0
epoch = 0
tf.summary.scalar('loss_av', average)
for step in max_step:
    #other stuff
    average = (average*(step-1)+loss)/step
    if step % steps_per_epoch == 0:
        epoch = epoch + 1
        av = sess.run(tf.constant(average))
        summary_writer.add_summary(av, epoch)