diff --git a/network.py b/network.py index 9b08e06..467c7fc 100644 --- a/network.py +++ b/network.py @@ -41,7 +41,7 @@ class Network: # self.learning_rate = 0.01 self.global_step = tf.Variable(0, trainable=False, name="global_step") - self.learning_rate = tf.maximum(self.min_learning_rate, tf.train.exponential_decay(self.max_learning_rate, self.global_step, 50000, 0.96, staircase=True), name="alpha") + self.learning_rate = tf.maximum(self.min_learning_rate, tf.train.exponential_decay(self.max_learning_rate, self.global_step, 50000, 0.96, staircase=True), name="learning_rate")