You don’t have to manage for each DataLoader , will manage batches for you! You just have to manage the input and transformation for one item of the Dataset, the loader will manage the rest for the Batch. That is the point of having them in the first place.