Custom Dataset Finetuning

[saransh03sharma]

Could you add some pointers on how to use your trained model for fine-tuning on a custom dataset? Also which model weights need to be stored where for the inferencing to work? @longyuewangdcu @minghao-wu @seeledu