Skip to content

A PyTorch implementation of GAN and VAE for CIFAR10 images.

License

Notifications You must be signed in to change notification settings

feroooooo/GAN-VAE-CIFAR10

Repository files navigation

GAN-VAE-CIFAR10

Environment

anaconda

$ conda env create -f env.yaml
$ conda activate gan_vae

pip

$ conda create -n gan_vae python=3.12.7
$ conda activate gan_vae
$ pip install -r requirements.txt

GAN

Train

Adjust the hyper-parmeters in train_gan.py and run it. Training information will be saved in directory: ./runs.

$ python train_gan.py

VAE

To Be Continued...

About

A PyTorch implementation of GAN and VAE for CIFAR10 images.

Resources

License

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages