2016-10-14 12 views
2

Tensorflowを使用してTensorflowの実行を視覚化しています。エポックごとに値を書き込むサマリーグラフを作成したいと思います。Tensorflow - サマリーをテンソルに書く

私はこのような何かをしたい:

with graph.as_default(): 
    tf_ending = tf.placeholder(tf.bool) 
    tf.scalar_summary('Loss', loss) # Some summaries are written every time 
    if tf_ending: 
     # This summary should only get written sometimes. 
     tf.scalar_summary('Total for Epoch', epoch_total) 

私はtf.merge_all_summaries()以外の何かを、別途サマリーのセットを管理するために必要な感覚を持っているが、私はそれがうまくいくかどうかはわかりません。

答えて

1

これを行う1つの方法は、カスタムサマリーprotobufをSummaryWriterに追加することです。

summary = tf.Summary() 
summary.value.add(tag='Total for Epoch',simple_value=epoch_total) 
summary_writer.add_summary(summary, train_step) 

これは、しかし、tensorflowグラフ(sess.run)を介して返される値(epoch_total)が必要:各エポック(セッション/グラフの外)の終わりに、あなたのような何かを追加することができます。また、これがこのようなことを行う最善の方法であるかどうかはわかりませんが、TFの例で使用されています。 hereおよびhere

関連する問題