after reading your document, i have no idea how to use VIbCReg in solo-learn to train custom dataset， can you show me an example? - Githubissues

vturrisi / solo-learn

solo-learn: a library of self-supervised methods for visual representation learning powered by Pytorch Lightning

MIT License

1.38k stars 181 forks source link

after reading your document, i have no idea how to use VIbCReg in solo-learn to train custom dataset， can you show me an example? #386

Closed RuojiWang closed 1 month ago

RuojiWang commented 1 month ago

after reading your document, i have no idea how to use VIbCReg in solo-learn to train custom dataset， can you show me an example?

vturrisi commented 1 month ago

Hey. You should replace the data part in: https://github.com/vturrisi/solo-learn/blob/b69b4bd27472593919956d9ac58902a301537a4d/scripts/pretrain/imagenet-100/vibcreg.yaml#L1-L56

with: https://github.com/vturrisi/solo-learn/blob/b69b4bd27472593919956d9ac58902a301537a4d/scripts/pretrain/custom/byol.yaml#L28-L37