Set learning rate to 10^-3

2019-12-16 10:31:29 -05:00 · 2019-12-16 10:31:29 -05:00 · e8e4dcec83
parent af59421dc0
commit e8e4dcec83
1 changed files with 1 additions and 1 deletions
--- a/train.py
+++ b/train.py
@ -61,7 +61,7 @@ scheduler = torch.optim.lr_scheduler.ReduceLROnPlateau(
    threshold=1e-8
 )

-while total_change >= 1e-4:
+while total_change >= 1e-3:
    # Shuffle training data and convert to one-hot encoding
    classes_dataset = classes_ordered[torch.randperm(len(classes_ordered))]
    onehot_dataset = util.messages_to_onehot(classes_dataset, order)