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
