pvae
pvae copied to clipboard
Negative KL loss
When I tested the Poincaré version on MNIST with latent_dim> 32 I had a negative KL loss, is this normal?
I also encounter negative KL loss when I just run the synthetic experiment following the instruction: python3 pvae/main.py --model tree --manifold PoincareBall --latent-dim 2 --hidden-dim 200 --prior-std 1.7 --c 1.2 --data-size 50 --data-params 6 2 1 1 5 5 --dec Wrapped --enc Wrapped --prior RiemannianNormal --posterior RiemannianNormal --epochs 1000 --save-freq 1000 --lr 1e-3 --batch-size 64 --iwae-samples 5000
Hi also got negative KL loss with the synthetic experiment after epoch 10. Did you find any solution?
I am also getting negative loss. I do not know why this is the case at low dimensions.