Large negative training loss in VAE

Yep just checked that was the problem there was a typo in my evaluate function. Thanks for the help !