Learn practical skills, build real-world projects, and advance your career
import torch
import torchvision
import numpy as np
import matplotlib.pyplot as plt
import torch.nn as nn
import torch.nn.functional as F
from torchvision.datasets import CIFAR10
from torchvision.transforms import ToTensor
from torchvision.utils import make_grid
from torch.utils.data.dataloader import DataLoader
from torch.utils.data import random_split
%matplotlib inline
dataset = torchvision.datasets.CIFAR100(root='data/',
                                         train=True,
                                         download=True,
                                         transform=ToTensor())


test_dataset = torchvision.datasets.CIFAR100(root='data/',
                                        train=False,
                                        download=True,
                                        transform=ToTensor())
Files already downloaded and verified Files already downloaded and verified
project_name = 'course5-cifar100'
dataset_size = len(dataset)
dataset_size
50000
test_dataset_size = len(test_dataset)
test_dataset_size
10000