tflearn learning_rate摘要

时间:2016-11-01 16:11:00

标签: tensorflow tensorboard

学习率自动添加到SGD和Momentum的摘要集合中。 但是它们不会添加到摘要中。获取此密钥并在TensorBoard上可视化的最快最简单的方法是什么?

1 个答案:

答案 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)