Training code

[liwenyang-911]

Dear author, thank you for your excellent work, is there a training code, please

[sebaxakerhtc]

@liwenyang-911 What you want to train? It works with any models...