vector-quantize-pytorch
本向量量化库来源于Deepmind的TensorFlow实现,并转化为Pytorch库,使用指数移动平均法来更新字典。它在高质量图像(如VQ-VAE-2)和音乐(如Jukebox)生成中已取得成功,支持多种残差VQ方法、代码簿初始化和正则化,显著提升了量化效果和稳定性。