zohaibmohammad

{"0": ["n01440764", "airplane"], "1": ["n01443537", "automobile"], "2": ["n01484850", "bird"], "3": ["n01491361", "cat"], "4": ["n01494475", "deer"], "5": ["n01496331", "dog"], "6": ["n01498041", "frog"], "7": ["n01514668", "horse"], "8": ["n01514859", "ship"], "9": ["n01518878", "Truck"]}

zohaibmohammad

#Import needed packages
import torch
import torch.nn as nn
from torchvision.datasets import CIFAR10
from torchvision.transforms import transforms
from torch.utils.data import DataLoader
from torch.optim import Adam
from torch.autograd import Variable
import numpy as np