AntixK/PyTorch-VAE
A collection of Variational Autoencoder implementations in PyTorch with reproducible training pipelines on CelebA dataset.

Velocity · 7d
+3.3
★ / day
Trend
→steady
star history
This repository provides PyTorch implementations of various VAE architectures including Beta-VAE, IWAE, VQ-VAE, WAE, and DFC-VAE. All models are trained consistently on the CelebA dataset to enable fair comparison. The project uses PyTorch Lightning for training infrastructure and emphasizes reproducibility through structured config files. Architecture choices closely follow the original papers where specified.