Quellcode durchsuchen

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

robrao vor 8 Jahren
Ursprung
Commit
92d751b4a9
1 geänderte Dateien mit 1 neuen und 1 gelöschten Zeilen
  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)