Hi, thank you for the elegant work that inspires me a lot.
I'm curious about the implementation of kl divergence loss since it seems quite different from plain VAE. The latent space represents as Gaussian param (μ, σ) in plain VAE, whereas the proposed represents it as a 8-dim vector. The difference makes me confused about how to compute kl loss, but i'm not able to find where this loss function locates.
Could anyone tell me where the part of KL divergence loss locates?
Hi, thank you for the elegant work that inspires me a lot.
I'm curious about the implementation of kl divergence loss since it seems quite different from plain VAE. The latent space represents as Gaussian param (μ, σ) in plain VAE, whereas the proposed represents it as a 8-dim vector. The difference makes me confused about how to compute kl loss, but i'm not able to find where this loss function locates.
Could anyone tell me where the part of KL divergence loss locates?