def make_loader()

in py/validate.py [0:0]

• 20 lines of code
• 2 McCabe index (conditional complexity)

def make_loader(batch_size, random=False, seed=1):
    valdir = '/home/jhj/imagenet/data/local/packages/ai-group.imagenet-full-size/prod/imagenet_full_size/val'

    normalize = transforms.Normalize(mean=[0.485, 0.456, 0.406],
                                     std=[0.229, 0.224, 0.225])

    dataset = datasets.ImageFolder(valdir, transforms.Compose([
        transforms.Resize(256),
        transforms.CenterCrop(224),
        transforms.ToTensor(),
        normalize,
    ]))

    sampler = None
    if random:
        sampler = torch.utils.data.RandomSampler(dataset)
        torch.manual_seed(seed)

    return torch.utils.data.DataLoader(
        dataset,
        sampler=sampler,
        batch_size=batch_size,
        shuffle=False,
        num_workers=0)