소스 검색

update tensorboard example

Aymeric Damien 9 년 전
부모
커밋
29c7bee7d3
2개의 변경된 파일51개의 추가작업 그리고 7개의 파일을 삭제
  1. 2 2
      examples/4_Utils/tensorboard_basic.py
  2. 49 5
      notebooks/4_Utils/tensorboard_basic.ipynb

+ 2 - 2
examples/4_Utils/tensorboard_basic.py

@@ -18,7 +18,7 @@ learning_rate = 0.01
 training_epochs = 25
 batch_size = 100
 display_step = 1
-logs_path = '/tmp/tensorflow_logs'
+logs_path = '/tmp/tensorflow_logs/example'
 
 # tf Graph Input
 # mnist data image of shape 28*28=784
@@ -61,7 +61,7 @@ with tf.Session() as sess:
     sess.run(init)
 
     # op to write logs to Tensorboard
-    summary_writer = tf.train.SummaryWriter(logs_path)
+    summary_writer = tf.train.SummaryWriter(logs_path, graph=tf.get_default_graph())
 
     # Training cycle
     for epoch in range(training_epochs):

파일 크기가 너무 크기때문에 변경 상태를 표시하지 않습니다.
+ 49 - 5
notebooks/4_Utils/tensorboard_basic.ipynb