pytorch-vq-vae

PyTorch implementation of VQ-VAE by Aäron van den Oord et al.

Requirements

$ pip3 install -r requirements.txt

Name		Name	Last commit message	Last commit date
Latest commit History 24 Commits
dataLoader		dataLoader
images		images
model		model
.gitignore		.gitignore
LICENSE		LICENSE
README.md		README.md
b2s_train_stl.py		b2s_train_stl.py
requirements.txt		requirements.txt
test_carton.py		test_carton.py
test_cifar.py		test_cifar.py
test_stl.py		test_stl.py
train_VQ-CLIP_ImgEncoder.py		train_VQ-CLIP_ImgEncoder.py
train_VQ-PixelCNN.py		train_VQ-PixelCNN.py
train_VQ-ResNet50.py		train_VQ-ResNet50.py
train_VQ-VQ-loss.py		train_VQ-VQ-loss.py
train_VQ-VQ.py		train_VQ-VQ.py
train_carton.py		train_carton.py
train_cifar.py		train_cifar.py
train_stl.py		train_stl.py
vq-vae.ipynb		vq-vae.ipynb