update LR scheduler

2018-11-05 09:20:18 +01:00 · 2018-11-05 09:20:18 +01:00 · 587097affb
parent 2ccf68cf96
commit 587097affb
1 changed files with 1 additions and 1 deletions
--- a/train.py
+++ b/train.py
@ -113,7 +113,7 @@ def main(opt):
        if epoch < 50:
            lr = 1e-3
        else:
-            lr = 1e-3
+            lr = 1e-4
        for g in optimizer.param_groups:
            g['lr'] = lr