explosion/spaCy
1
1
Fork 0
mirror of https://github.com/explosion/spaCy.git synced 2025-09-19 18:42:37 +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 # scalar loss
# loss += xp.sum(log_norm - log_marg) # 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 offset += ll
return loss, gradients return loss, gradients