def get_imagenet()

in vision/data.py [0:0]

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

def get_imagenet(datapath, split, batch_size, shuffle, transform=TRANSFORM):
    ds = torchvision.datasets.ImageNet(root=datapath, split=split, transform=transform)
    loader = torch.utils.data.DataLoader(ds, shuffle=shuffle, batch_size=batch_size)
    return ds, loader