Request for Training Code

[angieAAAAA]

Thank you for providing the inference code and making this project available!

I was wondering if it would be possible to share the training code as well. Having access to the training code would help me better understand the full pipeline and allow for further experimentation or fine-tuning on custom data.