glenn-jocher commited on
Commit
c9a3b14
·
unverified ·
1 Parent(s): 3eefab1

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

Browse files
Files changed (1) hide show
  1. utils/datasets.py +1 -1
utils/datasets.py CHANGED
@@ -522,7 +522,7 @@ class LoadImagesAndLabels(Dataset):
522
  self.im_hw0, self.im_hw = [None] * n, [None] * n
523
  fcn = self.cache_images_to_disk if cache_images == 'disk' else self.load_image
524
  results = ThreadPool(NUM_THREADS).imap(fcn, range(n))
525
- pbar = tqdm(enumerate(results), total=n, bar_format=BAR_FORMAT)
526
  for i, x in pbar:
527
  if cache_images == 'disk':
528
  gb += self.npy_files[i].stat().st_size
 
522
  self.im_hw0, self.im_hw = [None] * n, [None] * n
523
  fcn = self.cache_images_to_disk if cache_images == 'disk' else self.load_image
524
  results = ThreadPool(NUM_THREADS).imap(fcn, range(n))
525
+ pbar = tqdm(enumerate(results), total=n, bar_format=BAR_FORMAT, disable=LOCAL_RANK > 0)
526
  for i, x in pbar:
527
  if cache_images == 'disk':
528
  gb += self.npy_files[i].stat().st_size