Train code release

[mehran-hosseinzadeh]

Hi and thanks for your great work!

You have mentioned that the training code is being cleaned up to be released. I would like to ask when this code (even part of it or its latest version) could be published.

Thanks again!

[fengxiuyaun]

？？？？