tensorboard

1
2
3
4
5
6
7
8
9
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
请作者喝一杯咖啡☕️