Update dataloaders.py

nikitabelooussovbtis · Jan 9, 2023 · 9cddd85 · 9cddd85
1 parent 4efaaa6
commit 9cddd85
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/utils/dataloaders.py b/utils/dataloaders.py
@@ -124,7 +124,7 @@ def create_dataloader(path,
             pad=pad,
             image_weights=image_weights,
             prefix=prefix)
-
+    print(dataset.labels,flush=True)
     batch_size = min(batch_size, len(dataset))
     nd = torch.cuda.device_count()  # number of CUDA devices
     nw = min([os.cpu_count() // max(nd, 1), batch_size if batch_size > 1 else 0, workers])  # number of workers