Skip to content

Commit

Permalink
thats all fine
Browse files Browse the repository at this point in the history
  • Loading branch information
Parry-Parry committed Nov 19, 2024
1 parent e376bd9 commit 911b578
Showing 1 changed file with 0 additions and 1 deletion.
1 change: 0 additions & 1 deletion rankers/modelling/dot.py
Original file line number Diff line number Diff line change
Expand Up @@ -185,7 +185,6 @@ def forward(self,

query_reps = self._encode_q(**queries) if queries is not None else None
docs_batch_reps = self._encode_d(**docs_batch) if docs_batch is not None else None
breakpoint()
pred, labels, inbatch_pred = self.prepare_outputs(query_reps, docs_batch_reps, labels)
inbatch_loss = self.inbatch_loss_fn(inbatch_pred, torch.eye(inbatch_pred.shape[0]).to(inbatch_pred.device)) if inbatch_pred is not None else 0.

Expand Down

0 comments on commit 911b578

Please sign in to comment.