/Conditional-Generative-Adversarial-Network

Tutorial on creating your own Conditional GAN in PyTorch

Primary LanguageJupyter Notebook

Conditional-Generative-Adversarial-Network

This repo contains a tutorial and an implementation of the Conditional Generative Adversarial Nets paper. The model is written in PyTorch and familiarity with the original Generative Adversarial Nets is assumed.

Usage

  1. Clone the repository
git clone https://github.com/bhiziroglu/Conditional-Generative-Adversarial-Network.git

  1. Download the CIFAR-10 dataset Python version and extract it in the same directory with this project.

  2. Run the following code to start training.

python main.py
  1. You can also look at the tutorial. To install Jupyter Notebook, take a look at their website.