Explorar o código

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

robrao %!s(int64=8) %!d(string=hai) anos
pai
achega
92d751b4a9
Modificáronse 1 ficheiros con 1 adicións e 1 borrados
  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)