def METHOD_NAME(self, dataset):
    return DataLoader(
        dataset, batch_size=1, # for bucketing
        shuffle=False, num_workers=self.datarc['num_workers'],
        drop_last=False, pin_memory=True, collate_fn=dataset.collate_fn
    )