diff --git a/deepspeech/training/trainer.py b/deepspeech/training/trainer.py index a14cd7a028061905e9d8e32a3ac51fe180a6a99e..6c18fa369a41ae61d6d5f3a65c7fe16edac65151 100644 --- 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}!")