resume not increase epoch and step, before_train do it

3 years ago · 9092594fab
parent ad6d09d622
commit 9092594fab
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.iteration = infos["step"]
-            self.epoch = infos["epoch"] + 1
+            self.epoch = infos["epoch"]
            scratch = False
            logger.info(
                f"Restore ckpt: epoch {self.epoch }, step {self.iteration}!")