GitHub repository: AndrewBoessen/VQ-VAE
Primary language: Python
Created: 2024-10-11T15:40:53Z
Description: A PyTorch implementation of Vector Quantized Variational Autoencoder (VQ-VAE) with EMA updates, pretrained encoder, and K-means initialization. Efficient discrete representation learning for various data types.