未验证 提交 6b525e00 编写于 作者: F Feng Ni 提交者: GitHub

[cherry-pick] revert det tqdm eval (#5620)

* revert det tqdm eval, test=document_fix

* revert det tqdm eval, test=document_fix
上级 00149853
......@@ -141,6 +141,10 @@ class LogPrinter(Callback):
dtime=str(data_time),
ips=ips)
logger.info(fmt)
if mode == 'eval':
step_id = status['step_id']
if step_id % 100 == 0:
logger.info("Eval iter: {}".format(step_id))
def on_epoch_end(self, status):
if dist.get_world_size() < 2 or dist.get_rank() == 0:
......
......@@ -501,7 +501,7 @@ class Trainer(object):
flops_loader = create('{}Reader'.format(self.mode.capitalize()))(
self.dataset, self.cfg.worker_num, self._eval_batch_sampler)
self._flops(flops_loader)
for step_id, data in enumerate(tqdm(loader)):
for step_id, data in enumerate(loader):
self.status['step_id'] = step_id
self._compose_callback.on_step_begin(self.status)
# forward
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册