diff --git a/mmdet/core/evaluation/eval_hooks.py b/mmdet/core/evaluation/eval_hooks.py index 140c1ed6304e796ba2b36330aa36e891bf6f946d..33af0f72c25faf70139c91959951770a145be9f0 100644 --- a/mmdet/core/evaluation/eval_hooks.py +++ b/mmdet/core/evaluation/eval_hooks.py @@ -34,7 +34,8 @@ class DistEvalHook(Hook): return runner.model.eval() results = [None for _ in range(len(self.dataset))] - prog_bar = mmcv.ProgressBar(len(self.dataset)) + if runner.rank == 0: + prog_bar = mmcv.ProgressBar(len(self.dataset)) for idx in range(runner.rank, len(self.dataset), runner.world_size): data = self.dataset[idx] data_gpu = scatter( @@ -48,8 +49,9 @@ class DistEvalHook(Hook): results[idx] = result batch_size = runner.world_size - for _ in range(batch_size): - prog_bar.update() + if runner.rank == 0: + for _ in range(batch_size): + prog_bar.update() if runner.rank == 0: print('\n')