The RAG Index / Embeddings / #47
KevinMusgrave/pytorch-metric-learning
by KevinMusgrave · Embeddings · updated today
Train and evaluate embeddings for similarity search with PyTorch. Includes contrastive losses, hard example mining, samplers, and retrieval metrics.
73
momentum
6,344
stars
663
forks
#47
rank
computer-visioncontrastive-learningdeep-learningdeep-metric-learningembeddingsimage-retrievalmachine-learningmetric-learningpytorchself-supervised-learning
View on GitHub →