Переглянути джерело

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

robrao 8 роки тому
батько
коміт
92d751b4a9
1 змінених файлів з 1 додано та 1 видалено
  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)