The process is described in Supplementary, but where do you see the code to train the model? The repository is the inference pipeline.