How to train the model

[ooFormalinoo]

Hi,thanks to your job.However,what should I do if I want to train the model myself instead of just using the pre-trained model ?

[sniklaus]

I am afraid that my time is limited and I do not have the resources to provide the training code. The provided implementation is fully differentiable though, which is why you just need to provide a data loader and implement a loss function. Feel free to fork this repository and see whether you can make it work. It would be nice to have a fork with a working training section that I could link to.