maxjcohen / vqvae

VQ-VAE implementation in pytorch, supporting EMA and Gumbel trainings. Applicable for images and time series.

Geek Repo

Github PK Tool