diff --git a/deepspeech/exps/u2/model.py b/deepspeech/exps/u2/model.py index a5cef15c8..811da39b5 100644 --- a/deepspeech/exps/u2/model.py +++ b/deepspeech/exps/u2/model.py @@ -199,8 +199,8 @@ class U2Trainer(Trainer): report("Rank", dist.get_rank()) report("epoch", self.epoch) report('step', self.iteration) - report('step/total', - (batch_index + 1) / len(self.train_loader)) + report('iter', batch_index + 1) + report('total',len(self.train_loader)) report("lr", self.lr_scheduler()) self.train_batch(batch_index, batch, msg) self.after_train_batch() diff --git a/deepspeech/training/trainer.py b/deepspeech/training/trainer.py index c1afd3629..9ff95f29b 100644 --- a/deepspeech/training/trainer.py +++ b/deepspeech/training/trainer.py @@ -247,8 +247,8 @@ class Trainer(): report("Rank", dist.get_rank()) report("epoch", self.epoch) report('step', self.iteration) - report('step/total', - (batch_index + 1) / len(self.train_loader)) + report('iter', batch_index + 1) + report('total',len(self.train_loader)) report("lr", self.lr_scheduler()) self.train_batch(batch_index, batch, msg) self.after_train_batch()