KL Term becomes too low during training
KL Term becomes too low during training