The easiest way to use deep metric learning in your application. Modular, flexible, and extensible. Written in PyTorch.
machine-learning
computer-vision
deep-learning
pytorch
metric-learning
image-retrieval
deep-metric-learning
self-supervised-learning
contrastive-learning
-
Updated
Jun 9, 2022 - Python
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.