search

vturrisi / solo-learn

solo-learn: a library of self-supervised methods for visual representation learning powered by Pytorch Lightning
MIT License
1.42k stars 185 forks source link

How can we use pretrained model for our image classification task? #339

Closed saifhassan closed 1 year ago

saifhassan commented 1 year ago

Hey @turian @Froskekongen @Borda @ojss @DonkeyShot21,

Thank you for this nice work. I have one query regarding down-stream task, I have pretrained model on my dataset (image classification) using main_pretrain.py with BYOL and simclr and others. Now I want to use this pretrained model for my actual task which is image classification? How can I use that, please guide me regarding this.

Like you have done for object detection in sub-repo, I want to perform for image classification.

Thank you in advance.

DonkeyShot21 commented 1 year ago

You tagged wrong again :)

I replied in the other thread https://github.com/vturrisi/solo-learn/issues/338

saifhassan commented 1 year ago

yeah again wrong but I have gone through your profile and github issues/replies. So I knew that you would give response quickly.

Thank you so much.