如何在tensorflow的重新训练最后一层使用摘要

时间:2016-05-10 17:28:07

标签: tensorflow tensorboard

我成功地使用tensorflow's tutorial on retraining最后一层来处理新类,我想添加一些摘要来检查交叉熵是如何演变的。

我调查了documentation并尝试复制它,但我没有成功。 现在,我可以启动张量板并查看图表,但其他选项卡上没有显示任何其他内容(事件,图像,直方图)。

是否有人在初始再培训示例中添加了摘要?

提前致谢

2 个答案:

答案 0 :(得分:2)

我将TensorBoard摘要添加到TensorFlow图像再培训tutorial的股票代码中。如果您想在本教程中使用TensorBoard,可以查看latest TensorFlow code on GitHub。如果您对更多细节感兴趣,我还会写一篇关于它的blog post

答案 1 :(得分:1)

要显示标量摘要(等),您需要执行以下四项操作:

  1. 在图表中添加摘要节点(例如tf.scalar_summarytf.histogram_summary
  2. 将所有摘要合并到一个操作tf.merge_all_summaries
  3. 将摘要操作作为训练循环的一部分运行,并存储输出
  4. 将输出的摘要字符串传递给tf.train.SummaryWriter
  5. 你在做所有这些步骤吗?