An Evaluation of Generative Adversarial Networks

An evaluation of recent variants of generative adversarial networks by PyTorch

Requirements

pip3 install -r requirements.txt

tensorboard --logdir=out/

CUDA_VISIBLE_DEVICES=0 python3 train.py --mode dcgan --data celeba --d_iters 1 --g_iters 2 --gpu --ttur

WGAN

CUDA_VISIBLE_DEVICES=0 python3 train.py --mode wgan --data celeba --d_iters 5 --g_iters 1 --gpu --ttur

LSGAN

CUDA_VISIBLE_DEVICES=0 python3 train.py --mode lsgan --data celeba --d_iters 1 --g_iters 1 --gpu --ttur

WGAN-GP

CUDA_VISIBLE_DEVICES=0 python3 train.py --mode wgan-gp --data celeba --d_iters 5 --g_iters 1 --gpu --ttur

LSGAN-GP

CUDA_VISIBLE_DEVICES=0 python3 train.py --mode lsgan-gp --data celeba --d_iters 1 --g_iters 1 --gpu --ttur

DRAGAN

CUDA_VISIBLE_DEVICES=0 python3 train.py --mode dragan --data celeba --d_iters 1 --g_iters 1 --gpu --ttur

GAN-QP-L1

CUDA_VISIBLE_DEVICES=0 python3 train.py --mode gan-qp-l1 --data celeba --d_iters 2 --g_iters 1 --gpu --ttur

GAN-QP-L2

CUDA_VISIBLE_DEVICES=0 python3 train.py --mode gan-qp-l2 --data celeba --d_iters 2 --g_iters 1 --gpu --ttur

./exp.sh [GPU_ID] [MODE] [DATA] [TTUR]

Name		Name	Last commit message	Last commit date
Latest commit History 31 Commits
LICENSE		LICENSE
README.md		README.md
custom_helpers.py		custom_helpers.py
data.py		data.py
eval.py		eval.py
eval_fid.py		eval_fid.py
eval_fid.sh		eval_fid.sh
eval_is.sh		eval_is.sh
exp.sh		exp.sh
frechet_inception_distance.py		frechet_inception_distance.py
helpers.py		helpers.py
inception.py		inception.py
inception_score.py		inception_score.py
make_video.py		make_video.py
nn.py		nn.py
requirements.txt		requirements.txt
test.py		test.py
torchsummary.py		torchsummary.py
train.py		train.py
vid.sh		vid.sh