From c1d3ff517a6b532be8cc07fd5fd152daee0ddaff Mon Sep 17 00:00:00 2001
From: Matthew Honnibal <honnibal+gh@gmail.com>
Date: Sun, 20 Aug 2017 14:42:23 +0200
Subject: [PATCH] Track loss in tagger

---
 spacy/pipeline.pyx | 4 ++++
 1 file changed, 4 insertions(+)

diff --git a/spacy/pipeline.pyx b/spacy/pipeline.pyx
index 634d3e4b5..8c3759778 100644
--- a/spacy/pipeline.pyx
+++ b/spacy/pipeline.pyx
@@ -294,6 +294,8 @@ class NeuralTagger(BaseThincComponent):
         doc.is_tagged = True
 
     def update(self, docs_tokvecs, golds, drop=0., sgd=None, losses=None):
+        if losses is not None and self.name not in losses:
+            losses[self.name] = 0.
         docs, tokvecs = docs_tokvecs
 
         if self.model.nI is None:
@@ -302,6 +304,8 @@ class NeuralTagger(BaseThincComponent):
         loss, d_tag_scores = self.get_loss(docs, golds, tag_scores)
 
         d_tokvecs = bp_tag_scores(d_tag_scores, sgd=sgd)
+        if losses is not None:
+            losses[self.name] += loss
         return d_tokvecs
 
     def get_loss(self, docs, golds, scores):