Code for training

[markkua]

Thanks for your great work and clean test code.

I would like to ask about your plan to release code for training since I would like to apply your method to other datasets.

[baegwangbin]

Hi, thanks for your interest and sorry for the delayed response.
I am a bit busy at the moment, but I would probably have some time in March or April to work on the training code release.

[nandometzger]

same here, I would also be interested to reproduce your results.

[baegwangbin]

Hi, sorry for the delay.
I have just added the training code.
Please let me know if there is anything unclear about the code.