Open mr-mateusz opened 2 months ago
Hi @mr-mateusz , can you try graphbolt https://docs.dgl.ai/stochastic_training/index.html, which is our latest SOTA GNN dataloader. DGL dataloader is in a unmaintained mode now and will be deprecated in the future.
If you observe same issue, please reachout.
🐛 Bug
DGL DataLoader does not maintain order of examples with shuffle=False when num_workers > 1 and batch_size * num_workers <= dataset_size. Elements in a single batch are in order but they are not in order across batches. This behavior seems inconsistent with the expected operation of a DataLoader when shuffle=False.
To Reproduce
Code:
Output:
Expected behavior
DataLoader to produces data in the same order as the input indices when shuffle=False, regardless of the number of workers or batch size.
Environment
conda
,pip
, source): pip install dgl -f https://data.dgl.ai/wheels/torch-2.3/repo.htmlAdditional context