標量摘要的張量板平均值

Question

我正在嘗試使用TensorBoard可視化我的輸出損耗值。 我的方法很簡單。 我只是在一個小批量的每次迭代之后添加一個summary。標量。 在一個紀元的最后，張量板將顯示每次迭代后的所有損失。 我想獲得所有這些迭代的平均值，這意味着我希望張量板為我顯示每個時期的一個損失值。

謝謝。

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)