Pytorch-metric-learning

Features

- Added a batch_size parameter to [CustomKNN](https://kevinmusgrave.github.io/pytorch-metric-learning/inference_models/#customknn). This computes k-nn per batch of query embeddings (using [BatchedDistance](https://kevinmusgrave.github.io/pytorch-metric-learning/distances/#batcheddistance)), which requires less memory than computing the entire distance matrix at once.

Bug Fixes

- 451 (thanks cwkeam)
- 453

Bug Fixes

- 447

New Loss Function: SubCenterArcFace

- [Documentation](https://kevinmusgrave.github.io/pytorch-metric-learning/losses/#subcenterarcfaceloss)
- [Example notebook](https://github.com/KevinMusgrave/pytorch-metric-learning/blob/master/examples/notebooks/SubCenterArcFaceMNIST.ipynb)
- [Paper](https://www.ecva.net/papers/eccv_2020/papers_ECCV/papers/123560715.pdf)
- [Issue](https://github.com/KevinMusgrave/pytorch-metric-learning/issues/208)
- [Pull Request](https://github.com/KevinMusgrave/pytorch-metric-learning/pull/424)

Thanks chingisooinar!

Bug fixes

- 427
- 428

Bug fixes
- 420
- 422

New features

[CentroidTripletLoss](https://kevinmusgrave.github.io/pytorch-metric-learning/losses/#centroidtripletloss)
Implementation of [On the Unreasonable Effectiveness of Centroids in Image Retrieval](https://arxiv.org/pdf/2104.13643.pdf)

[VICRegLoss](https://kevinmusgrave.github.io/pytorch-metric-learning/losses/#vicregloss)
Implementation of [VICReg: Variance-Invariance-Covariance Regularization for Self-Supervised Learning](https://arxiv.org/pdf/2105.04906.pdf)

[AccuracyCalculator](https://kevinmusgrave.github.io/pytorch-metric-learning/accuracy_calculation/)
- Added mean reciprocal rank as an accuracy metric. Available as "mean_reciprocal_rank".
- Added return_per_class argument for AccuracyCalculator. This is like avg_of_avgs but returns the accuracy per class, instead of averaging them for you.


Related issues
369
372
374
394

Contributors
Thanks to cwkeam and mlw214!