Add VQ-VAE pretrained model

facebookresearch / NeuralCompression

A collection of tools for neural compression enthusiasts.

MIT License

500 stars 44 forks source link

Add VQ-VAE pretrained model #218

Closed mmuckley closed 1 year ago

mmuckley commented 1 year ago

This PR includes a few bug fixes for the VQ-VAE model, as well as adding a pretrained VQ-VAE architecture for reproducing the training from the paper.

This PR also includes a number of bug fixes for the GAN training code to match with the open-source ILLM implementation.