Where is the codebook loss?

[waynchi]

Hi! I'm going through the codebase trying to understand the various losses that you use. I can find the recons and commit losses, but I can't find the codebook loss anywhere. Could someone point me to where that would be?

Thanks!

[MlWoo]

@waynchi The paper explains it very clear. "To speed up training, the loss of codebook instead uses EMA up- dates over the codebook variables."