提交 bab29b94 编写于 作者: H Hui Zhang

fix train log

上级 56e55c21
...@@ -199,8 +199,8 @@ class U2Trainer(Trainer): ...@@ -199,8 +199,8 @@ class U2Trainer(Trainer):
report("Rank", dist.get_rank()) report("Rank", dist.get_rank())
report("epoch", self.epoch) report("epoch", self.epoch)
report('step', self.iteration) report('step', self.iteration)
report('step/total', report('iter', batch_index + 1)
(batch_index + 1) / len(self.train_loader)) report('total',len(self.train_loader))
report("lr", self.lr_scheduler()) report("lr", self.lr_scheduler())
self.train_batch(batch_index, batch, msg) self.train_batch(batch_index, batch, msg)
self.after_train_batch() self.after_train_batch()
......
...@@ -247,8 +247,8 @@ class Trainer(): ...@@ -247,8 +247,8 @@ class Trainer():
report("Rank", dist.get_rank()) report("Rank", dist.get_rank())
report("epoch", self.epoch) report("epoch", self.epoch)
report('step', self.iteration) report('step', self.iteration)
report('step/total', report('iter', batch_index + 1)
(batch_index + 1) / len(self.train_loader)) report('total',len(self.train_loader))
report("lr", self.lr_scheduler()) report("lr", self.lr_scheduler())
self.train_batch(batch_index, batch, msg) self.train_batch(batch_index, batch, msg)
self.after_train_batch() self.after_train_batch()
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册