smallflyingpig/pytorch_examples

some examples by pytorch

pytorch_examples

Note: go to wiki for loss curve and sample result

install the environment

conda env create -f environment.yml

gan_plain

GAN model according to Generative Adversarial Nets

mnist classification

classification model according the example of pytorch

vae model

VAE model according to Auto-encoding variational bayes

dcgan

DCGAN model according to unsupervised representation learning with deep convolutional generative adversarial

improved dcgan

improved DCGAN model accdording to improved Techniques for Training GANs

WGAN

WGAN model according Wasserstein GAN

WGAN-GP

WGAN-GP model according to Improved Training of Wasserstein GAN

LSGAN

LSGAN model according to least square generative adversarial net

BEGAN

BEGAN model according to BEGAN:Boundary Equilibrium Generative Adversarial Networks

InfoGAN

InfoGAN model according to InfoGAN: Interpretable Representation Learning by Information Maximizing Generative Adversarial Nets