The step to do inference is similar to training but with a few modification in configuration file for each stage:

Taking Embedding for example, while you read your config under LightningModule folder:

checkpoint_path: <Change the stored checkpoint_path to your best ckpt file instead>

Then,

#under GNNforLRT repository

traintrack ./configs/YOURPIPELINE --inference

And that’s it!