From 5177f3e7a0215c20895472f1e57137476ee649cb Mon Sep 17 00:00:00 2001 From: Glenn Jocher Date: Fri, 9 Nov 2018 16:48:55 +0100 Subject: [PATCH] updates --- train.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/train.py b/train.py index d935fb17..a1a443dd 100644 --- 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