pvae icon indicating copy to clipboard operation
pvae copied to clipboard

Negative KL loss

Open ManalHamzaoui opened this issue 4 years ago • 3 comments

When I tested the Poincaré version on MNIST with latent_dim> 32 I had a negative KL loss, is this normal?

ManalHamzaoui avatar Jan 03 '21 23:01 ManalHamzaoui

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

xinyue96 avatar Mar 10 '21 23:03 xinyue96

Hi also got negative KL loss with the synthetic experiment after epoch 10. Did you find any solution?

saniazahan avatar Jul 26 '21 10:07 saniazahan

I am also getting negative loss. I do not know why this is the case at low dimensions.

omiethescientist avatar Mar 27 '22 00:03 omiethescientist