Learn practical skills, build real-world projects, and advance your career
Created 4 years ago
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)