Skip to content

Commit 7c6088f

Browse files
author
Yuheng Tu
committed
change sign
1 parent c202b15 commit 7c6088f

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

src/helm/benchmark/reeval_runner.py

+1-1
Original file line numberDiff line numberDiff line change
@@ -82,7 +82,7 @@ def _estimate_model_ability(
8282
) -> float:
8383
def closure():
8484
optim.zero_grad()
85-
probs = torch.sigmoid(ability - difficulties)
85+
probs = torch.sigmoid(ability + difficulties)
8686
loss = -torch.distributions.Bernoulli(probs=probs).log_prob(responses).mean()
8787
loss.backward()
8888
return loss

0 commit comments

Comments
 (0)