Hello author, I want to reproduce your non-pre-trained version, do you have a training code, or how can I do it, thank you very much.