diff --git a/ppdet/engine/callbacks.py b/ppdet/engine/callbacks.py index 1c906320b64ece9c3a7a27399e8b8e2f0cb8c039..77ca94602c518411f040fa9e55f6c2ee656d726f 100644 --- a/ppdet/engine/callbacks.py +++ b/ppdet/engine/callbacks.py @@ -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: diff --git a/ppdet/engine/trainer.py b/ppdet/engine/trainer.py index cda3e6823827fcd951f07733f6950209ccc32606..35e1b97102bf3c1650a9907f33d0d938b0288c60 100644 --- a/ppdet/engine/trainer.py +++ b/ppdet/engine/trainer.py @@ -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