search

sthalles / SimCLR

PyTorch implementation of SimCLR: A Simple Framework for Contrastive Learning of Visual Representations
https://sthalles.github.io/simple-self-supervised-learning/
MIT License
2.19k stars 458 forks source link

how to do downstream task for image classification #35

Open ramdhan1989 opened 3 years ago

ramdhan1989 commented 3 years ago

Hi, is there any example how to do fine tuning / downstream task for classification by adding fc after the backbone ?

thanks

fjbriones commented 3 years ago

I believe their notebook implementation is already suited for that task.