When will you publish the train script?

[starmemda]

Hello! Thank you for the code, and I want to train the model by myself to prove the effectiveness of your method, when will you publish the train script?.

@CryhanFang