updates

2018-11-15 00:57:15 +01:00 · 2018-11-15 00:57:15 +01:00 · 1ea87c49c4
commit 1ea87c49c4
parent a17280ac72
1 changed files with 2 additions and 2 deletions
--- a/train.py
+++ b/train.py
@ -105,9 +105,9 @@ def main(opt):

        # Update scheduler (manual)  at 0, 54, 61 epochs to 1e-3, 1e-4, 1e-5
        if epoch > 50:
-            lr = 1e-5
-        else:
            lr = 1e-4
+        else:
+            lr = 1e-3
        for g in optimizer.param_groups:
            g['lr'] = lr