Hi, are you asking about this one .
_all__ = ['AlexNet', 'alexnet']


data_transforms = transforms.Compose([
    transforms.Resize(256),
    transforms.CenterCrop(224),
    transforms.ToTensor(),
    transforms.Normalize(mean=[0.485, 0.456, 0.406], std=[0.229, 0.224, 0.225]),
])


val_dirr = "~/Alexnet/val"

model_urls = {
    'alexnet': 'https://download.pytorch.org/models/alexnet-owt-4df8aa71.pth',
}

#train_dataset = torchvision.datasets.ImageFolder(root=..., transform=data_transforms, target_transform=None, is_valid_file=None)
test_dataset  = torchvision.datasets.ImageFolder(root=val_dirr, transform=data_transforms, is_valid_file=None)

test_loader = torch.utils.data.DataLoader(dataset=test_dataset,  batch_size=100, shuffle=True)

#If making subsets
#test_ds  = torch.utils.data.Subset(test_loader.dataset, range(0,2))
#test_loader_1 = torch.utils.data.DataLoader(dataset=test_ds,  batch_size=1, shuffle=True)