notebooks icon indicating copy to clipboard operation
notebooks copied to clipboard

How to use self supervised learning of the ViTMSN model in image_classification.ipynb?

Open JensSettelmeier opened this issue 1 year ago • 0 comments

Dear Niels and huggingface team,

thanks for the notebook about fine-tuning ViTs from huggingface. Similar to ViTMAE, ViTMSN should be trainable in self-supervised fashion. That means it should be possible to load a pre-trained model with: pretrained_ViT = AutoModel.from_pretrained("facebook/vit-msn-base-4") and then finetune it on a costume unlabeled big dataset with the masked siamese approach, right? I thought there should be a function like "ViTMSNForPreTraining," similar to "ViTMAEForPreTraining," but unfortunately, I couldn't find it. Is there a workaround, or is someone from huggingface currently working on it by any chance?

I would appreciate your support!

Cheers, Jens

JensSettelmeier avatar Feb 06 '23 18:02 JensSettelmeier