cerberus icon indicating copy to clipboard operation
cerberus copied to clipboard

Possible to finetune the model on custom dataset(s)

Open ajinkya-kulkarni opened this issue 2 years ago • 1 comments

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

ajinkya-kulkarni avatar Dec 07 '23 11:12 ajinkya-kulkarni

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.

simongraham avatar Nov 27 '24 19:11 simongraham