├── .gitignore ├── README.md ├── datasets.py ├── final_project.pdf ├── functions.py ├── modules.py ├── pixelcnn_baseline.py ├── pixelcnn_prior.py ├── samples ├── samples_CIFAR10.png ├── samples_FashionMNIST.png ├── samples_MNIST.png ├── vae_reconstructions_CIFAR10.png ├── vae_reconstructions_FashionMNIST.png ├── vae_reconstructions_MNIST.png ├── vae_samples_CIFAR10.png ├── vae_samples_FashionMNIST.png ├── vae_samples_MNIST.png ├── vqvae_reconstructions_CIFAR10.png ├── vqvae_reconstructions_FashionMNIST.png └── vqvae_reconstructions_MNIST.png ├── test_functions.py ├── vae.py └── vqvae.py /.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ritheshkumar95/pytorch-vqvae/HEAD/.gitignore -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ritheshkumar95/pytorch-vqvae/HEAD/README.md -------------------------------------------------------------------------------- /datasets.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ritheshkumar95/pytorch-vqvae/HEAD/datasets.py -------------------------------------------------------------------------------- /final_project.pdf: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ritheshkumar95/pytorch-vqvae/HEAD/final_project.pdf -------------------------------------------------------------------------------- /functions.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ritheshkumar95/pytorch-vqvae/HEAD/functions.py -------------------------------------------------------------------------------- /modules.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ritheshkumar95/pytorch-vqvae/HEAD/modules.py -------------------------------------------------------------------------------- /pixelcnn_baseline.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ritheshkumar95/pytorch-vqvae/HEAD/pixelcnn_baseline.py -------------------------------------------------------------------------------- /pixelcnn_prior.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ritheshkumar95/pytorch-vqvae/HEAD/pixelcnn_prior.py -------------------------------------------------------------------------------- /samples/samples_CIFAR10.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ritheshkumar95/pytorch-vqvae/HEAD/samples/samples_CIFAR10.png -------------------------------------------------------------------------------- /samples/samples_FashionMNIST.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ritheshkumar95/pytorch-vqvae/HEAD/samples/samples_FashionMNIST.png -------------------------------------------------------------------------------- /samples/samples_MNIST.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ritheshkumar95/pytorch-vqvae/HEAD/samples/samples_MNIST.png -------------------------------------------------------------------------------- /samples/vae_reconstructions_CIFAR10.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ritheshkumar95/pytorch-vqvae/HEAD/samples/vae_reconstructions_CIFAR10.png -------------------------------------------------------------------------------- /samples/vae_reconstructions_FashionMNIST.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ritheshkumar95/pytorch-vqvae/HEAD/samples/vae_reconstructions_FashionMNIST.png -------------------------------------------------------------------------------- /samples/vae_reconstructions_MNIST.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ritheshkumar95/pytorch-vqvae/HEAD/samples/vae_reconstructions_MNIST.png -------------------------------------------------------------------------------- /samples/vae_samples_CIFAR10.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ritheshkumar95/pytorch-vqvae/HEAD/samples/vae_samples_CIFAR10.png -------------------------------------------------------------------------------- /samples/vae_samples_FashionMNIST.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ritheshkumar95/pytorch-vqvae/HEAD/samples/vae_samples_FashionMNIST.png -------------------------------------------------------------------------------- /samples/vae_samples_MNIST.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ritheshkumar95/pytorch-vqvae/HEAD/samples/vae_samples_MNIST.png -------------------------------------------------------------------------------- /samples/vqvae_reconstructions_CIFAR10.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ritheshkumar95/pytorch-vqvae/HEAD/samples/vqvae_reconstructions_CIFAR10.png -------------------------------------------------------------------------------- /samples/vqvae_reconstructions_FashionMNIST.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ritheshkumar95/pytorch-vqvae/HEAD/samples/vqvae_reconstructions_FashionMNIST.png -------------------------------------------------------------------------------- /samples/vqvae_reconstructions_MNIST.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ritheshkumar95/pytorch-vqvae/HEAD/samples/vqvae_reconstructions_MNIST.png -------------------------------------------------------------------------------- /test_functions.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ritheshkumar95/pytorch-vqvae/HEAD/test_functions.py -------------------------------------------------------------------------------- /vae.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ritheshkumar95/pytorch-vqvae/HEAD/vae.py -------------------------------------------------------------------------------- /vqvae.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ritheshkumar95/pytorch-vqvae/HEAD/vqvae.py --------------------------------------------------------------------------------