How to do reconstruction on CelebA flow model?

Hi, thanks for your brilliant work! As far as I know, Flow++ is the best generative flow model on images up until now.

I have difficulty starting up with your code and what I want to know most is how to reconstruct a real image and get the internal representation out of the model. Can you show me a demo script to start with or give some hint on how to start with?

Much appreciated.