issues
search
qdrant
/
quaterion
Blazing fast framework for fine-tuning similarity learning models
https://quaterion.qdrant.tech/
Apache License 2.0
633
stars
45
forks
source link
Feature: FastAP Loss Function
#200
Closed
KarahanS
closed
1 year ago
KarahanS
commented
1 year ago
[x]
FastAP Loss
function will be implemented.
[x] Test for the new loss function will be implemented.
Reference implementations:
https://github.com/kunhe/FastAP-metric-learning/blob/master/pytorch/FastAP_loss.py
https://github.com/KevinMusgrave/pytorch-metric-learning/blob/691a6354be130547a8e26e170b86cf65c36cd791/src/pytorch_metric_learning/losses/fast_ap_loss.py
KarahanS
commented
1 year ago
Please refer to the PR #199.
KarahanS
commented
1 year ago
FastAP loss function is implemented.
Unittest is prepared - in which we compare the result of our loss function with that of the official implementation.
Loss function is implemented to comply with
device
consistency and tested using gpu.
Reference implementations: