PyTorch Metric Learning

by   Kevin Musgrave, et al.

Deep metric learning algorithms have a wide variety of applications, but implementing these algorithms can be tedious and time consuming. PyTorch Metric Learning is an open source library that aims to remove this barrier for both researchers and practitioners. The modular and flexible design allows users to easily try out different combinations of algorithms in their existing code. It also comes with complete train/test workflows, for users who want results fast. Code and documentation is available at


page 1

page 2

page 3

page 4


metric-learn: Metric Learning Algorithms in Python

metric-learn is an open source Python package implementing supervised an...

Match Cutting: Finding Cuts with Smooth Visual Transitions

A match cut is a transition between a pair of shots that uses similar fr...

Metric Learning and Adaptive Boundary for Out-of-Domain Detection

Conversational agents are usually designed for closed-world environments...

Visual Microfossil Identificationvia Deep Metric Learning

We apply deep metric learning for the first time to the prob-lem of clas...

Dynamic Metric Learning: Towards a Scalable Metric Space to Accommodate Multiple Semantic Scales

This paper introduces a new fundamental characteristic, , the dynamic ra...

A Metric Learning Reality Check

Deep metric learning papers from the past four years have consistently c...

PyTorch Adapt

PyTorch Adapt is a library for domain adaptation, a type of machine lear...

Please sign up or login with your details

Forgot password? Click here to reset