Commit 92d751b4 authored by robrao's avatar robrao Committed by Neal Wu
Browse files

correct tf.summary.scalar illegal name `learning rate`

parent ceed1a31
......@@ -128,7 +128,7 @@ class ResNet(object):
def _build_train_op(self):
"""Build training specific ops for the graph."""
self.lrn_rate = tf.constant(self.hps.lrn_rate, tf.float32)
tf.summary.scalar('learning rate', self.lrn_rate)
tf.summary.scalar('learning_rate', self.lrn_rate)
trainable_variables = tf.trainable_variables()
grads = tf.gradients(self.cost, trainable_variables)
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment