ritheshkumar95 / Pytorch Vqvae
Vector Quantized VAEs - PyTorch Implementation
Stars: ✭ 396
Programming Languages
python
139335 projects - #7 most used programming language
Labels
Projects that are alternatives of or similar to Pytorch Vqvae
VAE-Gumbel-Softmax
An implementation of a Variational-Autoencoder using the Gumbel-Softmax reparametrization trick in TensorFlow (tested on r1.5 CPU and GPU) in ICLR 2017.
Stars: ✭ 66 (-83.33%)
Mutual labels: vae
Copycat-abstractive-opinion-summarizer
ACL 2020 Unsupervised Opinion Summarization as Copycat-Review Generation
Stars: ✭ 76 (-80.81%)
Mutual labels: vae
Neural Ode
Jupyter notebook with Pytorch implementation of Neural Ordinary Differential Equations
Stars: ✭ 335 (-15.4%)
Mutual labels: vae
contiguous-succotash
Recurrent Variational Autoencoder with Dilated Convolutions that generates sequential data implemented in pytorch
Stars: ✭ 71 (-82.07%)
Mutual labels: vae
disent
🧶 Modular VAE disentanglement framework for python built with PyTorch Lightning ▸ Including metrics and datasets ▸ With strongly supervised, weakly supervised and unsupervised methods ▸ Easily configured and run with Hydra config ▸ Inspired by disentanglement_lib
Stars: ✭ 41 (-89.65%)
Mutual labels: vae
S Vae Pytorch
Pytorch implementation of Hyperspherical Variational Auto-Encoders
Stars: ✭ 255 (-35.61%)
Mutual labels: vae
variational-autoencoder-theano
Variational Autoencoders (VAEs) in Theano for Images and Text
Stars: ✭ 54 (-86.36%)
Mutual labels: vae
Tensorflow Generative Model Collections
Collection of generative models in Tensorflow
Stars: ✭ 3,785 (+855.81%)
Mutual labels: vae
srVAE
VAE with RealNVP prior and Super-Resolution VAE in PyTorch. Code release for https://arxiv.org/abs/2006.05218.
Stars: ✭ 56 (-85.86%)
Mutual labels: vae
Pytorch rvae
Recurrent Variational Autoencoder that generates sequential data implemented with pytorch
Stars: ✭ 332 (-16.16%)
Mutual labels: vae
learning-to-drive-in-5-minutes
Implementation of reinforcement learning approach to make a car learn to drive smoothly in minutes
Stars: ✭ 227 (-42.68%)
Mutual labels: vae
ladder-vae-pytorch
Ladder Variational Autoencoders (LVAE) in PyTorch
Stars: ✭ 59 (-85.1%)
Mutual labels: vae
Daisyrec
A developing recommender system in pytorch. Algorithm: KNN, LFM, SLIM, NeuMF, FM, DeepFM, VAE and so on, which aims to fair comparison for recommender system benchmarks
Stars: ✭ 280 (-29.29%)
Mutual labels: vae
sqair
Implementation of Sequential Attend, Infer, Repeat (SQAIR)
Stars: ✭ 96 (-75.76%)
Mutual labels: vae
Dsprites Dataset
Dataset to assess the disentanglement properties of unsupervised learning methods
Stars: ✭ 340 (-14.14%)
Mutual labels: vae
generative deep learning
Generative Deep Learning Sessions led by Anugraha Sinha (Machine Learning Tokyo)
Stars: ✭ 24 (-93.94%)
Mutual labels: vae
classifying-vae-lstm
music generation with a classifying variational autoencoder (VAE) and LSTM
Stars: ✭ 27 (-93.18%)
Mutual labels: vae
Pytorch Rl
This repository contains model-free deep reinforcement learning algorithms implemented in Pytorch
Stars: ✭ 394 (-0.51%)
Mutual labels: vae
Pycadl
Python package with source code from the course "Creative Applications of Deep Learning w/ TensorFlow"
Stars: ✭ 356 (-10.1%)
Mutual labels: vae
Reproducing Neural Discrete Representation Learning
Course Project for IFT 6135 - Representation Learning
Project Report link: final_project.pdf
Instructions
- To train the VQVAE with default arguments as discussed in the report, execute:
python vqvae.py --data-folder /tmp/miniimagenet --output-folder models/vqvae
- To train the PixelCNN prior on the latents, execute:
python pixelcnn_prior.py --data-folder /tmp/miniimagenet --model models/vqvae --output-folder models/pixelcnn_prior
Datasets Tested
Image
- MNIST
- FashionMNIST
- CIFAR10
- Mini-ImageNet
Video
- Atari 2600 - Boxing (OpenAI Gym) code
Reconstructions from VQ-VAE
Top 4 rows are Original Images. Bottom 4 rows are Reconstructions.
MNIST
Fashion MNIST
Class-conditional samples from VQVAE with PixelCNN prior on the latents
MNIST
Fashion MNIST
Comments
- We noticed that implementing our own VectorQuantization PyTorch function speeded-up training of VQ-VAE by nearly 3x. The slower, but simpler code is in this commit.
- We added some basic tests for the vector quantization functions (based on
pytest). To run these tests
py.test . -vv
Authors
- Rithesh Kumar
- Tristan Deleu
- Evan Racah
Note that the project description data, including the texts, logos, images, and/or trademarks,
for each open source project belongs to its rightful owner.
If you wish to add or remove any projects, please contact us at [email protected].