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 
Accidentally deleted it
This commit is contained in:
Paul O'Leary McCann 2021-07-10 22:45:25 +09:00
parent c25ec292a9
commit 447c7070e3
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

@ -312,7 +312,7 @@ class CoreferenceResolver(TrainablePipe):
log_norm = ops.softmax(cscores, axis=1) log_norm = ops.softmax(cscores, axis=1)
grad = log_norm - log_marg grad = log_norm - log_marg
gradients.append((grad, cidx)) gradients.append((grad, cidx))
total_loss += float(loss) total_loss += float((grad ** 2).sum())
offset = hi offset = hi