Skip to content

Latest commit

 

History

History
31 lines (21 loc) · 1.22 KB

README.md

File metadata and controls

31 lines (21 loc) · 1.22 KB

GAN-Experiment

News

An anime branch for anime generation is now AVAILABLE!!! It was developed by my friend hsj.

Usage

python main.py --checkpoint_dir checkpoint --model dcgan --dataset MNIST --device 0

TO DO

  • Add support for CelebA and LSUN dataset.
  • Tensorboard to visualize the training process and generated images
  • Multi-GPU training

Reference

  • A good tutorial for GAN beginners.
  • The calculation of Inception Score is borrowed from this repository.
  • The calculation of FID is borrowed from this repository.
  • The implementation of self-attention is borrowed from this repository, and the download script for CelebA and LSUN download.sh is also borrowed from it.