tensorboard

Posted on Views: Views: 
1
2
3
4
5
6
7
8
9
10
fw = tf.summary.FileWriter(LOGS_DIR) # LOGS_DIR should correspond to the path you want to save the summaries in

summaries_op = tf.summary.merge_all()

for i in range(NUM_ITR):
    _, summaries_str = sess.run([train_op, summaries_op])
    fw.add_summary(summaries_str, global_step=i)

tensorboard --logdir=LOGS_DIR