learning rate won't initial from 0.001 when continuing training from checkpoint

2022-04-14 15:08:08 +08:00 · 2022-04-14 15:08:08 +08:00 · 29124c9458
commit 29124c9458
parent 255eec296a
1 changed files with 3 additions and 1 deletions
--- 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: