#568 · Primary category: Education & Research

generative-models

gan generative-model machine-learning pytorch rbm restricted-boltzmann-machine tensorflow vae

Collection of generative models, e.g. GAN, VAE in Pytorch and Tensorflow.

Project last updated:03/24/24

GitHub Stars

7.5K

Forks

2.0K

Contributors

7

License

Unlicense

Why we included this project

Reading a paper on GANs is one thing; seeing the same model written out in clean, runnable code is another. This repository collects working implementations of the major generative model families in PyTorch and TensorFlow. The GAN lineup alone spans the classics, from vanilla and Wasserstein to InfoGAN, DiscoGAN, and boundary equilibrium, and it continues with variational autoencoders, restricted Boltzmann machines, and a Helmholtz machine. Each model comes with training code and saves generated samples to an output folder, so you can run an experiment and inspect what it produces. That makes it a solid study companion for a student learning the mechanics or a researcher revisiting a classic paper, and a handy way to understand a generative architecture before picking a production tool. The code is meant for learning, not deployment, so expect to adapt it rather than run it as-is.

Articles for this project

No articles for this project yet.

To suggest a topic or contribute an article, contact us.

Related projects in this category