Training code

Thank you very much for developing and sharing such a wonderful tool. It has been incredibly helpful.

If possible, could you kindly consider releasing the training code used to reproduce the pre-trained model?  
Having access to the training pipeline would be very valuable for research and further applications.

Thank you for your time and consideration!