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.
1 parent 8f903c8 commit 1f924b2Copy full SHA for 1f924b2
cebra/models/criterions.py
@@ -95,7 +95,7 @@ def infonce(
95
96
Note:
97
- The behavior of this function changed beginning in CEBRA 0.3.0.
98
- The InfoNCE implementation is numerically stabilized.
+ The InfoNCE implementation is numerically stabilized.
99
"""
100
with torch.no_grad():
101
c, _ = neg_dist.max(dim=1, keepdim=True)
0 commit comments