PyTorch Metric Learning
- URL: http://arxiv.org/abs/2008.09164v1
- Date: Thu, 20 Aug 2020 19:08:56 GMT
- Title: PyTorch Metric Learning
- Authors: Kevin Musgrave, Serge Belongie, Ser-Nam Lim
- Abstract summary: 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, for users who want results fast.
- Score: 37.03614011735927
- License: http://arxiv.org/licenses/nonexclusive-distrib/1.0/
- Abstract: 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
https://www.github.com/KevinMusgrave/pytorch-metric-learning.
Related papers
- torchgfn: A PyTorch GFlowNet library [56.071033896777784]
torchgfn is a PyTorch library that aims to address this need.
It provides users with a simple API for environments and useful abstractions for samplers and losses.
arXiv Detail & Related papers (2023-05-24T00:20:59Z) - SequeL: A Continual Learning Library in PyTorch and JAX [50.33956216274694]
SequeL is a library for Continual Learning that supports both PyTorch and JAX frameworks.
It provides a unified interface for a wide range of Continual Learning algorithms, including regularization-based approaches, replay-based approaches, and hybrid approaches.
We release SequeL as an open-source library, enabling researchers and developers to easily experiment and extend the library for their own purposes.
arXiv Detail & Related papers (2023-04-21T10:00:22Z) - Trieste: Efficiently Exploring The Depths of Black-box Functions with
TensorFlow [50.691232400959656]
Trieste is an open-source Python package for Bayesian optimization and active learning.
Our library enables the plug-and-play of popular models within sequential decision-making loops.
arXiv Detail & Related papers (2023-02-16T17:21:49Z) - imitation: Clean Imitation Learning Implementations [7.7064239657103375]
imitation provides open-source implementations of imitation and reward learning algorithms in PyTorch.
We include three inverse reinforcement learning (IRL) algorithms, three imitation learning algorithms and a preference comparison.
arXiv Detail & Related papers (2022-11-22T03:11:29Z) - Xplique: A Deep Learning Explainability Toolbox [5.067377019157635]
We have developed Xplique: a software library for explainability.
It includes representative explainability methods as well as associated evaluation metrics.
The code is licensed under the MIT license and is freely available.
arXiv Detail & Related papers (2022-06-09T10:16:07Z) - Geometric Optimisation on Manifolds with Applications to Deep Learning [6.85316573653194]
We design and implement a Python library to help the non-expert using all these powerful tools.
The algorithms implemented in this library have been designed with usability and GPU efficiency in mind.
arXiv Detail & Related papers (2022-03-09T15:20:07Z) - Solo-learn: A Library of Self-supervised Methods for Visual
Representation Learning [83.02597612195966]
solo-learn is a library of self-supervised methods for visual representation learning.
Implemented in Python, using Pytorch and Pytorch lightning, the library fits both research and industry needs.
arXiv Detail & Related papers (2021-08-03T22:19:55Z) - Neko: a Library for Exploring Neuromorphic Learning Rules [0.3499870393443268]
Neko is a modular library for neuromorphic learning algorithms.
It can replicate state-of-the-art algorithms and, in one case, lead to significant outperformance in accuracy and speed.
Neko is an open source Python library that supports PyTorch and backends.
arXiv Detail & Related papers (2021-05-01T18:50:32Z) - Rethinking Few-Shot Image Classification: a Good Embedding Is All You
Need? [72.00712736992618]
We show that a simple baseline: learning a supervised or self-supervised representation on the meta-training set, outperforms state-of-the-art few-shot learning methods.
An additional boost can be achieved through the use of self-distillation.
We believe that our findings motivate a rethinking of few-shot image classification benchmarks and the associated role of meta-learning algorithms.
arXiv Detail & Related papers (2020-03-25T17:58:42Z) - PHOTONAI -- A Python API for Rapid Machine Learning Model Development [2.414341608751139]
PHOTONAI is a high-level Python API designed to simplify and accelerate machine learning model development.
It functions as a unifying framework allowing the user to easily access and combine algorithms from different toolboxes into custom algorithm sequences.
arXiv Detail & Related papers (2020-02-13T10:33:05Z)
This list is automatically generated from the titles and abstracts of the papers in this site.
This site does not guarantee the quality of this site (including all information) and is not responsible for any consequences.