浏览代码

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