Learn practical skills, build real-world projects, and advance your career
Updated 4 years ago
!pip install jovian --upgrade --quietWARNING: Retrying (Retry(total=4, connect=None, read=None, redirect=None, status=None)) after connection broken by 'NewConnectionError('<pip._vendor.urllib3.connection.VerifiedHTTPSConnection object at 0x7f2c887d4c10>: Failed to establish a new connection: [Errno -3] Temporary failure in name resolution')': /simple/jovian/
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 FashionMNIST
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 inlineproject_name = "trying-fashion-feedforward"dataset = FashionMNIST(root="download/", download=True, train=True, transform=ToTensor())
test_dataset = FashionMNIST(root="download/", download=True, train=False, transform=ToTensor())val_size = 10000
train_size = len(dataset) - val_size
train_dataset, val_dataset = random_split(dataset, [train_size, val_size])
len(train_dataset), len(val_dataset)(50000, 10000)