From 29124c94582acd611161c950b0a0d168c0de82c2 Mon Sep 17 00:00:00 2001
From: blessyyyu <954793264@qq.com>
Date: Thu, 14 Apr 2022 15:08:08 +0800
Subject: [PATCH] learning rate won't initial from 0.001 when continuing
 training from checkpoint

---
 kws/bin/train.py | 4 +++-
 1 file changed, 3 insertions(+), 1 deletion(-)

diff --git a/kws/bin/train.py b/kws/bin/train.py
index c7b092a..ab051ba 100644
--- a/kws/bin/train.py
+++ b/kws/bin/train.py
@@ -157,7 +157,9 @@ def main():
         infos = {}
     start_epoch = infos.get('epoch', -1) + 1
     cv_loss = infos.get('cv_loss', 0.0)
-
+    # get the last epoch lr
+    lr_last_epoch = infos.get('lr', 0.001)
+    configs['optim_conf']['lr'] = lr_last_epoch
     model_dir = args.model_dir
     writer = None
     if rank == 0: