Spaces:
Configuration error
Configuration error
Update losses.py
Browse files- 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):
|