学习率自动添加到SGD和Momentum的摘要集合中。 但是它们不会添加到摘要中。获取此密钥并在TensorBoard上可视化的最快最简单的方法是什么?
答案 0 :(得分:0)
如果您使用merge_all_summaries
,则会包含GraphKeys.SUMMARIES
中的所有摘要。这样做的一个例子是models/image/cifar10/cifar10_train.py
summary_op = tf.merge_all_summaries()
summary_writer = tf.train.SummaryWriter(FLAGS.train_dir, sess.graph)
summary_str = sess.run(summary_op)
summary_writer.add_summary(summary_str, step)