yocabon commited on
Commit
b1b5578
·
unverified ·
1 Parent(s): 0e7bc28

Update losses.py

Browse files
Files changed (1) hide show
  1. mast3r/losses.py +0 -6
mast3r/losses.py CHANGED
@@ -274,12 +274,6 @@ class InfoNCE(MatchingCriterion):
274
 
275
  class APLoss (MatchingCriterion):
276
  """ AP loss.
277
-
278
- Input: (N, M) values in [min, max]
279
- label: (N, M) values in {0, 1}
280
-
281
- Returns: 1 - mAP (mean AP for each n in {1..N})
282
- Note: typically, this is what you wanna minimize
283
  """
284
 
285
  def __init__(self, nq='torch', min=0, max=1, euc=False, **kw):
 
274
 
275
  class APLoss (MatchingCriterion):
276
  """ AP loss.
 
 
 
 
 
 
277
  """
278
 
279
  def __init__(self, nq='torch', min=0, max=1, euc=False, **kw):