Possible to finetune the model on custom dataset(s)

[ajinkya-kulkarni]

Hello team, thanks for this repo! I have a short question - is it possible to finetune the model on custom dataset(s)?
For example, if I have a custom annotated dataset (say similar to the CONIC dataset), is finetuning possible?

Best, Ajinkya

[simongraham]

Yes, you will be able to fine-tune if you take the model definition along with the pretrained weights from this repo. However, we don't provide the functionality to do that 'out-of-the-box' within this code-base.