cifar.py 227 Bytes
Newer Older
Soumith Chintala's avatar
Soumith Chintala committed
1
2
3
4
5
6
7
8
9
10
11
12
import torch
import torchvision.datasets as dset

print('\n\nCifar 10')
a = dset.CIFAR10(root="abc/def/ghi", download=True)

print(a[3])

print('\n\nCifar 100')
a = dset.CIFAR100(root="abc/def/ghi", download=True)

print(a[3])