Ich versuche, meine Ausgabeverlustwerte zu visualisierenmit TensorBoard. Die Art, wie ich es tue, ist unkompliziert. Ich füge einfach nach jeder Iteration über einen Minibatch eine summary.scalar hinzu. Am Ende einer Epoche zeigt Tensorboard alle Verluste nach jeder Iteration. Ich möchte den Durchschnitt all dieser Iterationen erhalten, was bedeutet, dass Tensorboard mir für jede Epoche einen Verlustwert anzeigt.
Vielen Dank.
Antworten:
1 für die Antwort № 1Versuchen Sie, einen anderen Skalar zu erstellen durchschnittlich und für jeden Schritt in einer Epoche berechnen Sie einen laufenden Durchschnitt wie folgt:
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)