File size: 268 Bytes
85e3d20
 
 
 
 
 
 
 
1
2
3
4
5
6
7
8
9
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)