优秘智能 · 内容中心

rag ·

pytorch-metric-learning

Train and evaluate embeddings for similarity search with PyTorch. Includes contrastive losses, hard example mining, samplers, and retrieval metrics.

原始来源:https://github.com/KevinMusgrave/pytorch-metric-learning