Jovian
Sign In
Learn practical skills, build real-world projects, and advance your career
4 years ago
import os
import torch
import torchvision
import tarfile
from torchvision.datasets.utils import download_url
from torch.utils.data import random_split
project_name = 'cnn-learn'
dataset_url = 'http://files.fast.ai/data/cifar10.tgz'
download_url(dataset_url, 'data/')
Downloading http://files.fast.ai/data/cifar10.tgz to data/cifar10.tgz 
HBox(children=(FloatProgress(value=1.0, bar_style='info', max=1.0), HTML(value='')))
with tarfile.open('data/cifar10.tgz', 'r:gz') as tar:
    tar.extractall(path='data/')
data_dir = 'data/cifar10'

print(os.listdir(data_dir))
['test', 'train', 'labels.txt']