Actually, this is a custom Dataset which can be used as an input to a DataLoader. If you are looking for custom DataLoader, usually the only parameter you can modify is collate_fn which corresponds to batching.