From e8e4dcec83f56d20735682e3ca03c61c6a790282 Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?Matt=C3=A9o=20Delabre?= Date: Mon, 16 Dec 2019 10:31:29 -0500 Subject: [PATCH] Set learning rate to 10^-3 --- train.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/train.py b/train.py index 64b6f7d..e4d0948 100644 --- 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)