Don't use tqdm in ranks!=0

2021-12-28 10:06:54 -07:00 · 2021-12-28 10:06:54 -07:00 · 93624fa4b2
commit 93624fa4b2
parent a12042ea99
1 changed files with 1 additions and 1 deletions
--- a/codes/train.py
+++ b/codes/train.py
@ -264,7 +264,7 @@ class Trainer:
            self.epoch = epoch
            if opt['dist']:
                self.train_sampler.set_epoch(epoch)
-            tq_ldr = tqdm(self.train_loader)
+            tq_ldr = tqdm(self.train_loader) if self.rank == 0 else self.train_loader

            _t = time()
            for train_data in tq_ldr: