From 676d160dfa50810b167694137d58d2a7797cbce6 Mon Sep 17 00:00:00 2001
From: Hui Zhang <zhtclz@foxmail.com>
Date: Sat, 18 Sep 2021 09:41:18 +0000
Subject: [PATCH] more resume ckpt info

---
 deepspeech/training/trainer.py | 6 +++++-
 1 file changed, 5 insertions(+), 1 deletion(-)

diff --git a/deepspeech/training/trainer.py b/deepspeech/training/trainer.py
index 7815ed67..22dded6f 100644
--- a/deepspeech/training/trainer.py
+++ b/deepspeech/training/trainer.py
@@ -184,11 +184,13 @@ class Trainer():
             self.iteration = infos["step"]
             self.epoch = infos["epoch"]
             scratch = False
+            logger.info(
+                f"Restore ckpt: epoch {self.epoch }, step {self.iteration}!")
         else:
             self.iteration = 0
             self.epoch = 0
             scratch = True
-        logger.info("Restore/Init checkpoint!")
+            logger.info("Init from scratch!")
         return scratch
 
     def maybe_batch_sampler_step(self):
@@ -207,6 +209,8 @@ class Trainer():
             # resume: train next_epoch and next_iteration
             self.epoch += 1
             self.iteration += 1
+            logger.info(
+                f"Resume train: epoch {self.epoch }, step {self.iteration}!")
 
         self.maybe_batch_sampler_step()