From 12d91ee9b89d317995c60cd386cd8ac2da4a0dfe Mon Sep 17 00:00:00 2001
From: Kai Chen <chenkaidev@gmail.com>
Date: Wed, 10 Apr 2019 22:13:55 -0700
Subject: [PATCH] only print the progress bar for rank 0
---
mmdet/core/evaluation/eval_hooks.py | 8 +++++---
1 file changed, 5 insertions(+), 3 deletions(-)
diff --git a/mmdet/core/evaluation/eval_hooks.py b/mmdet/core/evaluation/eval_hooks.py
index 140c1ed..33af0f7 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')
--
GitLab