Skip to content

rmdluo/GAN-Number-Generation

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

11 Commits
 
 
 
 
 
 
 
 
 
 
 
 

Repository files navigation

Pytorch-GAN

This is an implementation of the papers Generative Adversarial Networks and Conditional Generative Adversarial Nets in PyTorch, done for personal enrichment. To get started, follow the setup instructions and then the run instructions.

Setup

  1. Install PyTorch from https://pytorch.org/ - Note: they removed conda support, but you can still install it using pip in a conda environment (that's what I did)
  2. pip install tqdm pillow pandas numpy matplotlib torchsummary

Run

For GAN:

  1. Adjust the hyperparameters near the top of train.py as needed
  2. python train.py

For Conditional GAN:

  1. Adjust the hyperparameters near the top of train_conditional.py as needed
  2. python train_conditional.py

About

No description, website, or topics provided.

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages