Introduction to DCGAN using PyTorch

dataset.py

import zipfile

!mkdir data_faces && wget https://s3-us-west-1.amazonaws.com/udacity-dlnfd/datasets/celeba.zip 
with zipfile.ZipFile("celeba.zip","r") as zip_ref:
  zip_ref.extractall("data_faces/")
  
from torch.utils.data import DataLoader
transform = transforms.Compose([
                               transforms.Resize((img_size,img_size)),
                               transforms.ToTensor()])

dataset = datasets.ImageFolder('data_faces', transform=transform)
data_loader = DataLoader(dataset,batch_size=batch_size,shuffle=True)