Skip to content
Snippets Groups Projects
Commit 12d91ee9 authored by Kai Chen's avatar Kai Chen
Browse files

only print the progress bar for rank 0

parent fd67644c
No related branches found
No related tags found
No related merge requests found
......@@ -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')
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment