← all repositories

AntixK/PyTorch-VAE

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

PyTorch-VAE
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.

heatdrop uses Google Analytics to see which pages get read — nothing else. Your call. How we handle data.