nakosung/VQ-VAE

VQ-VAE implementation / pytorch

VQ-VAE(Neural Discrete Representation Learning)

https://arxiv.org/abs/1711.00937

Requirements

nsml

How to run

nsml run -d CelebA_128 -v -i

Introduction

This is a repro of Vector Quantisation VAE from Deepmind. Authors had applied VQ-VAE for various tasks, but this repo is a slight modification of yunjey's VAE-GAN(CelebA dataset) to replace VAE with VQ-VAE.