未验证 提交 2ee33c0d 编写于 作者: F Feng Ni 提交者: GitHub

revert det tqdm eval (#5619)

* revert det tqdm eval, test=document_fix

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