nakosung / VQ-VAE

VQ-VAE implementation / pytorch

Home Page:https://arxiv.org/abs/1711.00937

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

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.

image

image

About

VQ-VAE implementation / pytorch

https://arxiv.org/abs/1711.00937


Languages

Language:Python 100.0%