delu.utils.data#

An addition to torch.utils.data.

Enumerate

Make a PyTorch dataset return indices in addition to items (like enumerate, but for datasets).

IndexDataset

A trivial dataset that yields indices back to user (useful for DistributedDataParallel (DDP)).