TensorFLow tensorBoard 總結

1.什麼是tensorboard：

tensorboard是可視化tensorflow 訓練過程一些參數數據的工具。

2.如何使用：

pip安裝tensorflow後，使用

python tensorboard.py --logdir='/path/to/logdir' 啓動tensorboard

3,如何在event文件中添加自己想要可視化的數據：

a.定義summary operation:

                         tf.scaler_summary:用來添加一些標量，比如 lr，loss，accuracy ，etc

                         tf.image_summary:用來添加一些進入graph的輸入圖片

                         tf.histogram_summary:用來統計激活分佈，梯度分佈，權重分佈

                         tf.audio_summary:

  比如：

tf.scalar_summary('標籤',想要記錄的變量)

b.定義一個op來將所有的summary operation 合併起來

merged = tf.merge_all_summaries()

c.使用graph初始化一個summary_writer

train_writer = tf.train.SummaryWriter(FLAGS.summaries_dir + '/train',sess.graph)

     

d.每隔n step將summary寫入

    summary, acc = sess.run([merged, accuracy], feed_dict=feed_dict(False))

    train_writer.add_summary(summary, i)