We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
2 parents 71aa465 + 7c46d28 commit 42afb6bCopy full SHA for 42afb6b
wildlife_tools/train/objective.py
@@ -37,7 +37,7 @@ class TripletLoss(nn.Module):
37
38
"""
39
40
- def __init__(self, margin: int = 0.2, mining: str = "seminard", distance: str = "l2_squared"):
+ def __init__(self, margin: int = 0.2, mining: str = "semihard", distance: str = "l2_squared"):
41
super().__init__()
42
if distance == "cosine":
43
distance = distances.CosineSimilarity()
0 commit comments