Collecting links Variational autoencoder implementation in keras with a custom training step Custom Training Using the Strategy Class. Withing a custom loss function, it's sometimes necessary to get the epoch number. Some reserch in determining, controlling the VAE loss function to avoid posterior collapse in Re-balancing Variational Autoencoder Loss for