From 93624fa4b22b18f3e4569c8272998cbec348a7f0 Mon Sep 17 00:00:00 2001
From: James Betker <jbetker@gmail.com>
Date: Tue, 28 Dec 2021 10:06:54 -0700
Subject: [PATCH] Don't use tqdm in ranks!=0

---
 codes/train.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/codes/train.py b/codes/train.py
index 9475aa66..ec29b996 100644
--- 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: