explosion/spaCy
1
1
Fork 0
mirror of https://github.com/explosion/spaCy.git synced 2025-07-19 20:52:23 +03:00
Code Issues Packages Projects Releases Wiki Activity

Transpose before calculating loss

Browse Source
This commit is contained in:
Paul O'Leary McCann 2021-06-03 22:18:14 +09:00
parent 18444fccd9
commit 67d9ebc922
1 changed files with 2 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
spacy/pipeline/coref.py
View File

@ -315,13 +315,11 @@ class CoreferenceResolver(TrainablePipe):
# do softmax to cscores
cscores = ops.softmax(cscores, axis=1)
diff = self.loss.get_grad(cscores, top_gscores)
diff = self.loss.get_grad(cscores.T, top_gscores.T).T
diff = diff[:, 1:]
gradients.append((diff, cidx))
# scalar loss
# loss += xp.sum(log_norm - log_marg)
loss += float(self.loss.get_loss(cscores, top_gscores))
loss += float(self.loss.get_loss(cscores.T, top_gscores.T))
offset += ll
return loss, gradients