Where is the training code?

[xiankgx]

Dear authors, I have read your paper

Lee, C.-H., Liu, Z., Wu, L., & Luo, P. (2019). MaskGAN: Towards Diverse and Interactive Facial Image Manipulation. http://arxiv.org/abs/1907.11922

and am interested to reimplement it in TensorFlow 2.0.

I'm having some issues with the VAE loss and would like to see how it is implemented but I can't seem to find the training code in this repo.

Most examples on the web on VAEs are on toy datasets like MNIST, and the loss consist of the binary cross entropy loss + KL divergence loss. However, since masks can have 19 classes, I'm not too certain if I can simply replace the binary cross entropy loss with the categorical cross entropy loss.

Please help.