Seems that p_z and p_x are swapped

[benmodels]

usually the latent variable (z) is uni modal and the data distribution (x) is multi-modal and GAN fails in finding some of the data modes. It's the other way around in your example. Was it intended?