Learn practical skills, build real-world projects, and advance your career
import torch
import torchvision
import numpy as np
from torchvision.datasets import FashionMNIST
from torchvision import transforms
data_transforms = transforms.Compose([
    transforms.RandomHorizontalFlip(),
    transforms.ToTensor(),
    transforms.Normalize(mean = (0.1, ), std = (0.1, )),
])
dataset = FashionMNIST(
    root = "data/",
    train=True,
    transform = data_transforms,
    download=True,
)
from torch.utils.data import random_split
lengths = [48000, 12000]
# lengths = [50000, 10000]
trian_set, valid_set = random_split(dataset = dataset, lengths = lengths)