Skip to content

Commit 2135bf2

Browse files
authored
minor simplification in solver
1 parent 276e5a3 commit 2135bf2

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

cebra/solver/single_session.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -120,8 +120,8 @@ def _inference(self, batch: cebra.data.Batch) -> cebra.data.Batch:
120120
"""
121121
batch.to(self.device)
122122
ref = self.model(batch.reference)
123-
neg = self.model(batch.negative)
124123
pos = self.model(batch.positive)
124+
neg = self.model(batch.negative)
125125
return cebra.data.Batch(ref, pos, neg)
126126

127127
def get_embedding(self, data: torch.Tensor) -> torch.Tensor:

0 commit comments

Comments
 (0)