Learn practical skills, build real-world projects, and advance your career
Updated 4 years ago
Classifying image of everyday object using feed foward network
!pip install jovian --upgrade --quiet
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
#Project name
project_name = 'feedforward-neural-network-modeling'