From 9092594fab69c6eecaaab0cfdc1f3c18e5ae9442 Mon Sep 17 00:00:00 2001
From: Hui Zhang <zhtclz@foxmail.com>
Date: Mon, 11 Oct 2021 07:27:43 +0000
Subject: [PATCH] resume not increase epoch and step, before_train do it

---
 deepspeech/training/trainer.py | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

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