Parcourir la source

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

robrao il y a 8 ans
Parent
commit
92d751b4a9
1 fichiers modifiés avec 1 ajouts et 1 suppressions
  1. 1 1
      resnet/resnet_model.py

+ 1 - 1
resnet/resnet_model.py

@@ -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)