def METHOD_NAME(self) -> det_torch.DataLoader:
    return det_torch.DataLoader(
        self.tokenized_datasets["train"],
        batch_size=self.context.get_per_slot_batch_size(),
        collate_fn=self.collator,
    )