2017-09-18 14 views
0

私はTensorBoardで出力損失値を視覚化しようとしています。私のやり方は簡単です。私はちょうどminibatchの各反復の後にsummary.scalarを追加します。 1エポックの終わりにテンソルボードは、各反復後にすべての損失を表示します。私はこれらすべての反復の平均を求めています。つまり、テンソルボードに各エポックごとに1つの損失値を表示することを意味します。スカラーサマリーのテンソルボード平均

ありがとうございます。

答えて

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)