lr floating division

ownway22 · Mar 29, 2017 · c4b48c4 · c4b48c4
1 parent 75e435f
commit c4b48c4
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/word_language_model/main.py b/word_language_model/main.py
@@ -174,7 +174,7 @@ def train():
             best_val_loss = val_loss
         else:
             # Anneal the learning rate if no improvement has been seen in the validation dataset.
-            lr /= 4
+            lr /= 4.0
 except KeyboardInterrupt:
     print('-' * 89)
     print('Exiting from training early')