Skip to content

nissmar/ARDM

Repository files navigation

ARDM

MVA Generative Models project on Order Agnostic Autoregressive Diffusion Models.

Face generation (CelebA)

img img imgimg img

Run TinyCelebA.ipynb to train and evaluate a UNet model on our resized Tiny CelebA dataset. The custom dataset is composed of 20.000 60×73 images with 32 grey-levels. This model was trained for 10 hours on a single GPU.

Character Generation (binary MNIST)

img img imgimg img

Run MNIST.ipynb to train and evaluate a UNet model on the binary MNIST dataset. The dataset is composed of 60.000 28x28 binary images. This model was trained for 1 hour.

Code reuse

We used parts from UNet, oardm and pytorch-fid

About

MVA generative models project

Resources

License

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published