Availability of training code

Hello, thank you for sharing this project. I‘m exploring using it with custom data and would need to perform training.

Could you please let me know if the code used for training the models in the paper is available? If it’s in a separate repository or branch, a pointer would be extremely helpful.

If it‘s not currently open-sourced, I understand completely. In that case, any additional documentation or hints about the training setup would also be very valuable.

Thanks for your work and any info you can provide.