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

Fix loss

Browse Source 
The loss was being returned as a single element array, which caused
training to die when it attempted to turn it into JSON.
This commit is contained in:
Paul O'Leary McCann 2021-05-21 15:46:50 +09:00
parent ff3fed06cf
commit e1b4a85bb9
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

@ -322,7 +322,7 @@ class CoreferenceResolver(TrainablePipe):
# scalar loss
# loss += xp.sum(log_norm - log_marg)
loss += self.loss.get_loss(cscores, top_gscores)
loss += float(self.loss.get_loss(cscores, top_gscores))
offset += ll
return loss, gradients