From fac81e46a71a803082828a5abc5efec8bf3adcad Mon Sep 17 00:00:00 2001 From: Hui Zhang Date: Wed, 22 Sep 2021 06:57:34 +0000 Subject: [PATCH] fix train log --- deepspeech/exps/u2/model.py | 4 ++-- deepspeech/training/trainer.py | 4 ++-- 2 files changed, 4 insertions(+), 4 deletions(-) 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()