Name		Name	Last commit message	Last commit date
parent directory ..
dcgan		dcgan
model		model
.DS_Store		.DS_Store
Readme.md		Readme.md
dcgan.py		dcgan.py
generate.py		generate.py

Readme.md

PyTorch Implementation of DCGAN

Prerequisites

PyTorch
torchvision

Training

Train on MNIST
```
python dcgan.py --cuda --dataset MNIST
```
Train on CIFAR
```
python dcgan.py --cuda --dataset CIFAR
```

Train on CPU: leave out the --cuda parameter

Generate

Generate MNIST

python generate.py --netG dcgan/MNIST.pth --batchSize 100 --dataset MNIST

Generate CIFAR

python generate.py --netG dcgan/CIFAR.pth --batchSize 100 --dataset CIFAR

Reference