카테고리 없음
Tensorboard로 텐서 그래프 보기
hs_kim
2020. 3. 10. 18:44
Graph 생성
파이썬 코드 안에 tensorflow.summary.FileWriter(logdir, tf.Session.graph) 를 사용
""" Model saver """
saver = tf.train.Saver(tf.global_variables(), max_to_keep=100)
config = tf.ConfigProto()
config.gpu_options.allow_growth = True
sess = tf.Session(config=config)
init = tf.global_variables_initializer()
sess.run(init)
# For the use of Tensorboard
writer = tf.summary.FileWriter("./graph/", sess.graph)
Tensorboard 실행
$ tensorboard --logdir ./graph/ --port=12345
Chrome을 통해 접속하면 아래와 같이 그래프를 볼 수 있다.
참고자료