vq-vae-2-pytorch
vq-vae-2-pytorch copied to clipboard
Update vqvae.py
trafficstars
correctly initialize the moving counter of codebook entry utilization.
initializing this to 0 leads to divides by ~0 and numerically unstable embeddings at the beginning of training.