How to train it?

Hi ! I’m really impressed by your model and would love to adapt it to non-natural images. Would it be possible to share the complete training code? Any work I build on it will cite both the code and the paper exactly as you specify.