Skip to content

Commit f368f5a

Browse files
authored
Merge pull request #6 from 573phn/patch-2
Added missing parenthesis to coref.py
2 parents 1ef3f3f + f33525f commit f368f5a

1 file changed

Lines changed: 1 addition & 1 deletion

File tree

src/coref.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -548,7 +548,7 @@ def train_doc(self, document):
548548

549549
# Negative marginal log-likelihood
550550
eps = 1e-8
551-
loss = torch.sum(torch.log(torch.sum(torch.mul(probs, gold_indexes), dim=1).clamp_(eps, 1-eps), dim=0) * -1
551+
loss = torch.sum(torch.log(torch.sum(torch.mul(probs, gold_indexes), dim=1).clamp_(eps, 1-eps), dim=0) * -1)
552552

553553
# Backpropagate
554554
loss.backward()

0 commit comments

Comments
 (0)