updates

2018-11-09 16:48:55 +01:00 · 2018-11-09 16:48:55 +01:00 · 5177f3e7a0
parent 664cbaab09
commit 5177f3e7a0
1 changed files with 1 additions and 1 deletions
--- a/train.py
+++ b/train.py
@ -120,7 +120,7 @@ def main(opt):

            # SGD burn-in
            if (epoch == 0) & (i <= 1000):
-                lr = 1e-3 * (i / 1000) ** 4
+                lr = 1e-4 * (i / 1000) ** 4
                for g in optimizer.param_groups:
                    g['lr'] = lr