from torchvision import datasets


taskname = "cifar10-training"
download_dir = "../env"

train_dataset = datasets.CIFAR10(root=f'{download_dir}/data', train=True, download=True)
test_dataset = datasets.CIFAR10(root=f'{download_dir}/data', train=False, download=True)