From 6e09224697a6481fdfe8233d1ca52bf1c9e2bf00 Mon Sep 17 00:00:00 2001
From: root <root@sp-mlc3-5423-0.mlc>
Date: Fri, 3 Jan 2020 00:38:08 +0100
Subject: [PATCH] change lr_decay config field to noam_schedule

---
 train.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/train.py b/train.py
index d24a4b6b..ed2085b9 100644
--- a/train.py
+++ b/train.py
@@ -602,7 +602,7 @@ def main(args):  # pylint: disable=redefined-outer-name
     if num_gpus > 1:
         model = apply_gradient_allreduce(model)
 
-    if c.lr_decay:
+    if c.noam_schedule:
         scheduler = NoamLR(optimizer,
                            warmup_steps=c.warmup_steps,
                            last_epoch=args.restore_step - 1)