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

Question on Class Loss #332

Closed kkokilep closed 1 year ago

kkokilep commented 1 year ago

Why does all the implementations seem to come with a class loss in addition to the loss associated with the method? I don't understand the reasoning for this. Any help with this issue is appreciated!

DonkeyShot21 commented 1 year ago

It is just for evaluation purposes, the linear layer receives features that are detached. See previous discussion: https://github.com/vturrisi/solo-learn/issues/230