search

AntixK / PyTorch-VAE

A Collection of Variational Autoencoders (VAE) in PyTorch.
Apache License 2.0
6.44k stars 1.05k forks source link

add reset+ema for vq-vae #82

Open SatireY opened 10 months ago

SatireY commented 10 months ago

I added EMAs and codebook reset techniques for VQ-VAE models, which enhanced the trained models.