标量摘要的张量板平均值

Question

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

谢谢。

Answer 1

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

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)

标量摘要的张量板平均值

tensorboard average of scalar summaries

python

tensorflow

tensorboard