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

Error using pre-trained model with CARD model #341

Closed saifhassan closed 1 year ago

saifhassan commented 1 year ago

Hello @DonkeyShot21 @vturrisi,

I want to train CARD Model (Classification Part) and use pre-trained model (generated by main_pretrain.py) as f_phi pretrained model so that it can improve accuracy.

What configuration settings, I will have to do so that I can use with CARD?

Thanks

vturrisi commented 1 year ago

I'm not familiar with the repo that you linked. After you pretrain a model with solo, you can use the weights saved in the checkpoint and load it accordingly. You will need to check how to integrate it afterwards.