Hello author, thank you for your excellent work. May I ask if you could share the loss function and training code? Thank you!