Training code pls

[Zhenyuan Zhang]

Hi @yuyi-sd , I have read your paper carefully and I‘m very interested in this part.
And I would appreciate it if you could share your code for training.
Looking forward to your reply~