2016-12-27 39 views
0

我是Tensorflow的新手,正在運行一個基本的CNN。作爲可視化培訓過程中的一種方式,我建立以便日後查看Tensorboard這樣,低損耗且精度摘要:在Tensorboard中複製的所有變量是什麼?

tf.summary.scalar("loss", cost) 
tf.summary.scalar("accuracy", accuracy) 

我初始化摘要如下。 (get_logdir_string()返回給定參數和當前日期時間組成的唯一的字符串)

merged_summary_op = tf.summary.merge_all() 
summary_writer = tf.summary.FileWriter(get_logdir_string('CIFAR10'), 
              graph=tf.get_default_graph()) 

然後對於每個小批量重複我這樣做:

_, summary = sess.run([optimizer, merged_summary_op], feed_dict={x: batch_x, 
                   y_true: batch_y, 
                   keep_prob: dropout}) 
summary_writer.add_summary(summary, step * batch_size) 

然後我跑Tensorboard和我提出了類似下面的內容,其中只有每個變量的第一個(沒有後綴)包含數據: Screenshot

有沒有人遇到過這個消息回覆?謝謝!

+0

看起來每個訓練步驟都有一個準確性/損失。你跑了多少個訓練步驟? 12?不應該「summary_writer.add_summary(summary,step * batch_size)」爲「summary_writer.add_summary(summary,step)」嗎? –

+0

這是全部包裝在一個循環中我有一個批量大小64, –

+0

出於某種原因,我不允許編輯我以前的評論 - 請忽略它,因爲我無意中提交了它。 @YaoZhang:這是全部包裝在 循環中,而step * batch_size

回答

1

我發現了罪魁禍首,並認爲我會在這裏發佈以供將來參考。

事實證明,我需要在每次運行前致電tf.reset_default_graph()

相關問題