pytorch-metric-learning
Subscription: Developer
Active
Subscription: Developer
A PyTorch library providing composable metric learning losses, miners, distances, samplers, and retrieval metrics for training embedding models, with support for contrastive, triplet, and proxy-based learning strategies.
pip install pytorch-metric-learning==3.0.2
3.0.2
Linux