Skip to content
This repository was archived by the owner on Dec 23, 2024. It is now read-only.

tsvikas/variational-autoencoder

Repository files navigation

variational-autoencoder

this is an implementation of variational autoencoder using pytorch-lightning. there are some helper classes for a lightning datamodule (with support for tran-validation split) and lightning model.

after training with a 2d latent space, the latent space looks like this: image of the latent space

About

a variational autoencoder for images

Topics

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published