lightly
lightly copied to clipboard
Add default parameters for all projection heads
Add default parameters for all projection heads
It's helpful to know what the default parameters were in the papers to get started. We should add the default projection head parameters which were used for pre-training on Imagenet to all projection and prediction heads in lightly/models/modules/heads.py
.
Hi, I would like to work on this issue. Pleas guide me further ^_^
Hi @himanshu007-creator that's great!
In case this is your first time contributing to an open source package, I would recommend you to take a look at our contribution guide. Let me know if you have any questions, I'm happy to help :)
Here is the list of implemented papers to get you started:
- Bootstrap your own latent: A new approach to self-supervised Learning, 2020
- Barlow Twins: Self-Supervised Learning via Redundancy Reduction, 2021
- SimSiam: Exploring Simple Siamese Representation Learning, 2020
- MoCo: Momentum Contrast for Unsupervised Visual Representation Learning, 2019
- SimCLR: A Simple Framework for Contrastive Learning of Visual Representations, 2020
- NNCLR: Nearest-Neighbor Contrastive Learning of Visual Representations, 2021
- SwaV: Unsupervised Learning of Visual Features by Contrasting Cluster Assignments, 2021
This has been implemented in #798