标量摘要的张量平均值

Question

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

谢谢。

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)