diff --git a/src/pytorch_metric_learning/losses/cosface_loss.py b/src/pytorch_metric_learning/losses/cosface_loss.py index 2371ed99..e18c3265 100644 --- a/src/pytorch_metric_learning/losses/cosface_loss.py +++ b/src/pytorch_metric_learning/losses/cosface_loss.py @@ -6,7 +6,7 @@ class CosFaceLoss(LargeMarginSoftmaxLoss): """ - Implementation of https://arxiv.org/pdf/1801.07698.pdf + Implementation of https://arxiv.org/abs/1801.09414 """ def __init__(self, *args, margin=0.35, scale=64, **kwargs):