Merge pull request #880 from PaddlePaddle/resume

resume not increase epoch and step, before_train do it
3 years ago · 8a35bfad31
parent ad6d09d622 9092594fab
commit 8a35bfad31
1 changed files with 2 additions and 2 deletions
--- a/deepspeech/training/trainer.py
+++ b/deepspeech/training/trainer.py
@ -199,8 +199,8 @@ class Trainer():
        if infos:
            # just restore ckpt
            # lr will resotre from optimizer ckpt
-            self.iteration = infos["step"] + 1
-            self.epoch = infos["epoch"] + 1
+            self.iteration = infos["step"]
+            self.epoch = infos["epoch"]
            scratch = False
            logger.info(
                f"Restore ckpt: epoch {self.epoch }, step {self.iteration}!")