Loading finetuned model from checkpoint for inference

[ogkdmr]

Hello @bfabiandev,

Thanks for open-sourcing this great codebase!

I pretrained a model for a regression task on my own dataset. I know have the checkpoints which I would like to deploy. I'm wondering how could we go about deploying the model in inference mode? Do you happen to have a prediction script?