Releasing the training code

Hi, thank for providing the inference code and dataset! It would be really helpful.
As mentioned in README, 

> An inference example is currently available. We are working on a release version of the training code, which will be available soon.

do you plan to release the training code soon?

Thank you in advance!

Best regards