Visualizing latent space in VAE and Discriminator

Hi,

Can anyone help me in visualizing a network that has Beta VAE, Discriminator, and Task Model?
I want to visualize the losses for each one of the modules and latent space for sample selection for some iterations.