I\'m working on Variational Autoencoder (VAE) and my model doesn\'t learn properly. The distribution in the latent vector is the same as for untrained model... (I\'m training mo