[pytorch] cifar10

-·2023년 12월 21일
>>> import torchvision
>>> import torchvision.transforms as transforms
>>> torchvision.datasets.CIFAR10(root='/data', train=True, download=False, transform=[transforms.ToTensor()])
>>> trainset.data.shape

(50000, 32, 32, 3)
profile
살아남은 자가 강한 것

0개의 댓글