Skip to content

AliceOfSNU/VQVAE2

Repository files navigation

VQVAE + PixelSNAIL prior

Trained VQVAE for 20 epochs + pixelSNAIL prior for 50 epochs, which took ~30m on RTX3080 laptop GPU. See config files for training details.

Generating Cat Faces

from top: epoch 0 -> bottom: epoch 50 cat_epochs

Conditional Generation of MNIST using PixelSNAIL

given a number between 0~9, generates MNIST-like image of that digit. PixelSNAIL trains much faster with conditioning.

mnist_conditional

Read more here (wandb)

About

experiments with vqvae + pixelsnail

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published