DCGAN LSGAN WGAN-GP DRAGAN PyTorch
News
GANs - Pytorch
Pytorch implementations of DCGAN, LSGAN, WGAN-GP(LP) and DRAGAN.
DCGAN | LSGAN | WGAN-GP | DRAGAN |
---|---|---|---|
DCGAN | LSGAN |
---|---|
WGAN-GP | DRAGAN |
WGAN-LP | DRAGAN-LP |
WGAN-GP | DRAGAN |
---|---|
Prerequisites
Datasets
Examples of training
Fashion-MNIST DCGAN
CUDA_VISIBLE_DEVICES=0 python train.py --dataset=fashion_mnist --epoch=25 --adversarial_loss_mode=gan
CelebA DRAGAN
CUDA_VISIBLE_DEVICES=0 python train.py --dataset=celeba --epoch=25 --adversarial_loss_mode=gan --gradient_penalty_mode=1-gp --gradient_penalty_sample_mode=dragan
Anime WGAN-GP
CUDA_VISIBLE_DEVICES=0 python train.py --dataset=anime --epoch=200 --adversarial_loss_mode=wgan --gradient_penalty_mode=1-gp --gradient_penalty_sample_mode=line --n_d=5
see more training exampls in commands.sh
tensorboard for loss visualization
tensorboard --logdir ./output/fashion_mnist_gan/summaries --port 6006