Changed alpha to learning_rate

This commit is contained in:
Alexander Munch-Hansen 2018-04-15 23:53:35 +02:00
parent 57fb1cb141
commit 611f6cdba0

View File

@ -41,7 +41,7 @@ class Network:
# self.learning_rate = 0.01 # self.learning_rate = 0.01
self.global_step = tf.Variable(0, trainable=False, name="global_step") 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")