Data.DrDataLoader

Click here to go back to the reference.

def DrDataLoader(dataset: DrDataset,
                 batch_size: int,
                 shuffle: bool,
                 follow_batch=None,
                 exclude_keys=None):
    """"""
    collate = DrCollate(follow_batch, exclude_keys)
    dataloader = DataLoader(dataset, batch_size=batch_size, shuffle=shuffle, collate_fn=collate)
    return dataloader