From 587097affb9fc1fe9892f063877eda7ce1a46998 Mon Sep 17 00:00:00 2001 From: Glenn Jocher Date: Mon, 5 Nov 2018 09:20:18 +0100 Subject: [PATCH] update LR scheduler --- train.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/train.py b/train.py index cb9eb39d..59ca365e 100644 --- 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