explosion/spaCy
1
1
Fork 0
mirror of https://github.com/explosion/spaCy.git synced 2025-09-18 18:12:45 +03:00
Code Issues Packages Projects Releases Wiki Activity

Fix backprop

Browse Source 
Training seems to actually run now!
This commit is contained in:
Paul O'Leary McCann 2021-05-18 20:09:27 +09:00
parent 2486b8ad4d
commit d22acee4f7
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
spacy/pipeline/coref.py
View File

@ -210,7 +210,7 @@ class CoreferenceResolver(TrainablePipe):
preds, backprop = self.model.begin_update(inputs) preds, backprop = self.model.begin_update(inputs)
score_matrix, mention_idx = preds score_matrix, mention_idx = preds
loss, d_scores = self.get_loss(examples, score_matrix, mention_idx) loss, d_scores = self.get_loss(examples, score_matrix, mention_idx)
backprop(d_scores) backprop((d_scores, mention_idx))
if sgd is not None: if sgd is not None:
self.finish_update(sgd) self.finish_update(sgd)