minor fix

cdluminate · Dec 6, 2023 · cf8adad · cf8adad
1 parent bfe932d
commit cf8adad
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/robrank/losses/infonce.py b/robrank/losses/infonce.py
@@ -15,7 +15,7 @@
 t=0.2  R@1 = 88.4
 '''
 
-def infonce(repA: th.Tensor, repB: th.Tensor, *, metric:str='C', t:float = 0.2) -> th.Tensor:
+def infonce(repA: th.Tensor, repB: th.Tensor, *, metric:str='C', t:float = 0.0) -> th.Tensor:
     # make sure shape is correct
     repA, repB = th.flatten(repA, 1), th.flatten(repB, 1)
     assert metric == 'C'