def cifar10_train(dataset_root, patch_size, patch_overlap): train_transform = amdim_transforms.TransformsC10Patches( patch_size=patch_size, patch_overlap=patch_overlap) dataset = CIFAR10Mixed(root=dataset_root, split='train', transform=train_transform, download=True) return dataset
def cifar10(dataset_root, patch_size, patch_overlap, split: str = 'train'): assert split in ('train', 'val') train_transform = amdim_transforms.TransformsC10Patches( patch_size=patch_size, patch_overlap=patch_overlap) dataset = CIFAR10Mixed( root=dataset_root, split=split, transform=train_transform, download=True, ) return dataset