Skip to content

Commit

Permalink
remove base loss from docs
Browse files Browse the repository at this point in the history
  • Loading branch information
ir2718 committed Oct 28, 2024
1 parent 285cea0 commit 31eeaaa
Showing 1 changed file with 2 additions and 2 deletions.
4 changes: 2 additions & 2 deletions docs/losses.md
Original file line number Diff line number Diff line change
Expand Up @@ -1252,9 +1252,10 @@ losses.SupConLoss(temperature=0.1, **kwargs)
## ThresholdConsistentMarginLoss
[Threshold-Consistent Margin Loss for Open-World Deep Metric Learning](https://arxiv.org/pdf/2307.04047){target=_blank}

This loss acts as a form of regularization and is usually combined with another metric loss function.

```python
losses.ThresholdConsistentMarginLoss(
base_loss,
lambda_plus=1.0,
lambda_minus=1.0,
margin_plus=0.9,
Expand All @@ -1267,7 +1268,6 @@ losses.ThresholdConsistentMarginLoss(

**Parameters**:

* **base_loss**: The final loss is calculated as `base_loss + tcm_loss`.
* **lambda_plus**: The scaling coefficient for the anchor-positive part of the loss. This is $\lambda^+$ in the above equation.
* **lambda_minus**: The scaling coefficient for the anchor-negative part of the loss. This is $\lambda^-$ in the above equation.
* **margin_plus**: The minimum anchor-positive similarity to be included in the loss. This is $m^+$ in the above equation.
Expand Down

0 comments on commit 31eeaaa

Please sign in to comment.