AI/모두를 위한 ML (SungKim)
[ML lab 09-2] Tensorboard (Neural Net for XOR)
ferozsun
2021. 2. 21. 01:27
TesnsorBoard
그래프를 통해 학습이 잘 이루어지고 있는지 한눈에 확인할 수 있다.
[5 steps of using TensorBoard]
1. 어떤 tensor를 loging할 것인지 정한다.
w2_hist = tf.summary.histogram("weights2", W2)
cost_summ = tf.summary.scalar("cost", cost)
2. merge 한다.
summary = tf.summary.merge_all()
3. file 위치를 적어 writer를 만들고 graph를 추가한다.
# Create summary writer
writer = tf.summary.FileWriter('./logs')
writer.add_graph(sess.graph)
4. summery를 실행시키고 writer에 추가한다.
s, _ = sess.run([summary, optimizer], feed_dict = feed_dict)
writer.add_summary(s, global_step = global_step)
5. 터미널에 디렉토리 위치와 명령어를 적어 TensorBoard 그래프를 확인할 수 있다.
$ tensorboard --logdir=./logs
histogram (multi-dimensional tensors)
데이터가 vector일 때 분포를 확인할 수 있다.
name_scope()
각 layer 별로 정리하여 graph hierarchy를 더 깔끔하게 확인할 수 있다.
Multiple runs
상위 디렉토리를 실행시키면 하위의 그래프들을 모두 동시에 실행시켜 비교할 수 있다.