From 2be4405b389909e1e5167b5059ab8e31bf39c42e Mon Sep 17 00:00:00 2001
From: blessyyyu <954793264@qq.com>
Date: Thu, 14 Apr 2022 15:49:41 +0800
Subject: [PATCH] fix intent bug with initial learning rate != 0.001

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

diff --git a/kws/bin/train.py b/kws/bin/train.py
index ab051ba..4d178c9 100644
--- a/kws/bin/train.py
+++ b/kws/bin/train.py
@@ -158,7 +158,7 @@ def main():
     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)
+    lr_last_epoch = infos.get('lr', configs['optim_conf']['lr'])
     configs['optim_conf']['lr'] = lr_last_epoch
     model_dir = args.model_dir
     writer = None