Skip to content

Commit

Permalink
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.
eek
Browse files Browse the repository at this point in the history
Parry-Parry committed Nov 26, 2024
1 parent 8fad54a commit 815cb1f
Showing 1 changed file with 1 addition and 0 deletions.
1 change: 1 addition & 0 deletions rankers/train/loss/__init__.py
Original file line number Diff line number Diff line change
@@ -189,6 +189,7 @@ def reg(self, reps, weight=0):
def forward(self, query_hidden_states, text_hidden_states, **kwargs):
q_reg = self.reg(query_hidden_states, self.q_weight)
d_reg = self.reg(text_hidden_states, self.d_weight)
self.step()
return q_reg + d_reg


0 comments on commit 815cb1f

Please sign in to comment.