Improved training of generative models