diff --git a/train.py b/train.py index 6250602..e6763c1 100644 --- a/train.py +++ b/train.py @@ -78,7 +78,7 @@ def get_learning_rate(batch): DECAY_STEP, # Decay step. DECAY_RATE, # Decay rate. staircase=True) - learing_rate = tf.maximum(learning_rate, 0.00001) # CLIP THE LEARNING RATE! + learning_rate = tf.maximum(learning_rate, 0.00001) # CLIP THE LEARNING RATE! return learning_rate def get_bn_decay(batch):