def METHOD_NAME(self):
    batch_size = self.config.get('runner.train_batch_size', None)
    data_loader = pgl.utils.data.Dataloader(
        self.train_ds,
        batch_size=batch_size,
        shuffle=True,
        num_workers=self.sample_workers,
        collate_fn=self.collate_fn)
    return data_loader