Play with Flemme

After installing flemme, you can test it with some small datasets to get an intuition about its capabilities.

Toy Example for Diffusion model

Configuration file: resources/toy_ddpm.yaml.

train_flemme --config resources/toy_ddpm.yaml
_images/ddpm_toy.png

MINST

Configuration files are in resources/img/mnist.

AutoEncoder & Variational AutoEncoder

_images/ae_mnist.png

Denoising Diffusion Probabilistic Model

_images/ddpm_mnist.png

CIFA10

Configuration files are in resources/img/cifar10.

AutoEncoder & Variational AutoEncoder

_images/ae_cifar10.png

Denoising Diffusion Probabilistic Model (conditional)

_images/cddpm_cifar10.png