Changed alpha to learning_rate
This commit is contained in:
parent
57fb1cb141
commit
611f6cdba0
|
@ -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")
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
|
Loading…
Reference in New Issue
Block a user