search

OML-Team / open-metric-learning

Metric learning and retrieval pipelines, models and zoo.
https://open-metric-learning.readthedocs.io/en/latest/index.html
Apache License 2.0
895 stars 61 forks source link

Metric learning without labels #628

Closed darthvaddor closed 1 month ago

darthvaddor commented 1 month ago

Is there a way to finetune dinov2 using OML with an unlabelled dataset. Does OML support that? Also, how will the STIR finetuning take place. My aim is to finetune dinov2 on my dataset, and also finetune the STIR. But all of this with an unlabelled dataset

darthvaddor commented 1 month ago

@AlekseySh @leoromanovich @b0nce @DaloroAT @dapladoc

I want this finetuned model to generate embeddings. Not for classification

AlekseySh commented 1 month ago

Hey, @JAVARSHA

We don't support unlabelled training in OML. I recommend you to find other solutions. You keywords are: Self Supervised Learning - SSL, Unsupervised Learning, Contrastive Learning. Check out this library: https://github.com/lightly-ai/lightly