Skip to content

Instantly share code, notes, and snippets.

View zohaibmohammad's full-sized avatar

Zohaib Muhammad zohaibmohammad

View GitHub Profile
#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