def prepare_dataloader()

in pai-python-sdk/training/pytorch_ddp/train_src/train_multinode.py [0:0]

• 8 lines of code
• 1 McCabe index (conditional complexity)

def prepare_dataloader(dataset: Dataset, batch_size: int):
    return DataLoader(
        dataset,
        batch_size=batch_size,
        pin_memory=True,
        shuffle=False,
        sampler=DistributedSampler(dataset),
    )