Disable pbar for DDP ranks > 0 (ultralytics#7440)

fhkiel-mlaip · Apr 16, 2022 · cccb3c0 · cccb3c0
1 parent fdbb96c
commit cccb3c0
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/utils/datasets.py b/utils/datasets.py
@@ -522,7 +522,7 @@ def __init__(self,
             self.im_hw0, self.im_hw = [None] * n, [None] * n
             fcn = self.cache_images_to_disk if cache_images == 'disk' else self.load_image
             results = ThreadPool(NUM_THREADS).imap(fcn, range(n))
-            pbar = tqdm(enumerate(results), total=n, bar_format=BAR_FORMAT)
+            pbar = tqdm(enumerate(results), total=n, bar_format=BAR_FORMAT, disable=LOCAL_RANK > 0)
             for i, x in pbar:
                 if cache_images == 'disk':
                     gb += self.npy_files[i].stat().st_size