def METHOD_NAME(self, dataset):
    return DataLoader(
        dataset, batch_size=self.datarc['train_batch_size'],
        shuffle=True, num_workers=self.datarc['num_workers'],
        collate_fn=dataset.collate_fn
    )