A modular, easy to use Encyclopedia of VAE

** STILL WORK IN PROGRESS **

This project aims at working from the repo Pytorch-VAE and upgrading it (a lot).

Features:

Packaging using modern tool like Poetry.
Start from the "vanilla" VAE model, and building upon it to get all the other ones: by changing the loss, some layers... The code is then far more easy to maintain than the original one.
Use the full power of Lightning tools, like LightningModule and especially the nice CLI.
Possibility to import any LightningDataModule to your project.

Name		Name	Last commit message	Last commit date
Latest commit History 140 Commits
.github/workflows		.github/workflows
configs		configs
encyclopedia_vae		encyclopedia_vae
tests		tests
.gitignore		.gitignore
.pre-commit-config.yaml		.pre-commit-config.yaml
.python-version		.python-version
Changelog.md		Changelog.md
LICENSE.md		LICENSE.md
README.md		README.md
pyproject.toml		pyproject.toml
ruff.toml		ruff.toml