Training the model

[dmariaa]

Hey there!

Thanks a lot for the awesome work and for sharing the code.

I'm interested in training your model from scratch using my own synthetic dataset. Do you have any plans to release the basic training code soon? Or maybe publishing some details about the training procedure?

Thanks again, and best regards!

[lpiccinelli-eth]

Hello @dmariaa, the training code is rolled out in the PR #76 , although not still in the main branch, you can try it out now.