resume not increase epoch and step, before_train do it

4 years ago · 493d6af232
parent b7b17e208b
commit 493d6af232
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}!")