Model training code

Hi,

Thanks for your amazing work! After the model passes the intern review, would you consider release the training code and the model architecture? This would be extremely helpful to the community, and provide invaluable insight for ones who want to train any novel model with their own data!

Thank you in advance!