cerberus
cerberus copied to clipboard
Possible to finetune the model on custom dataset(s)
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
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.