Skip to content

ahn1340/GAN

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

19 Commits
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 

Repository files navigation

GAN

Pytorch implementation of the DCGAN paper and Improved Techniques for Training GANs. Currently, only feature matching is implemented. Minibatch discrimination, label smoothing and historical averaging will be implemented soon.

Demo

CelebA dataset (vanilla adversarial loss)

Alt Text

CelebA dataset (feature matching loss)

Alt Text

How to use

Put your images in a folder and specify path to that folder in config/hyperparameters.yaml. Also, other training configurations can be modified in the file. Then run python main.py.

Training progress will be logged in WandB.

TODO

  • Minibatch discrimination
  • Label smoothing
  • Historical averaging

About

GAN implementation with Pytorch

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages