Tensorboard Durchschnitt der skalaren Zusammenfassungen

Question

Ich versuche meine Ausgangsverlustwerte mit TensorBoard zu visualisieren. Die Art, wie ich es mache, ist einfach. Ich füge nach jeder Iteration über ein Minibatch einfach ein summary.scalar hinzu. Am Ende einer Epoche wird Tensorboard mir alle Verluste nach jeder Iteration zeigen. Ich möchte den Durchschnitt aller dieser Iterationen erhalten, was bedeutet, dass ich möchte, dass das Tensorboard mir für jede Epoche einen Verlustwert anzeigt.

Vielen Dank.

Answer 1

versuchen Sie einen anderen skalaren Durchschnitt und für jeden Schritt in einer Epoche zu schaffen berechnen ein gleitender Mittelwert wie folgt aus:

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)