From a95eea4c061f9c52b660efe63b059a46cccc0ceb Mon Sep 17 00:00:00 2001 From: Matthew Honnibal Date: Thu, 13 Sep 2018 14:08:55 +0200 Subject: [PATCH] Fix multi-task objective for parser --- spacy/pipeline.pyx | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/spacy/pipeline.pyx b/spacy/pipeline.pyx index eb3894969..7b4572db5 100644 --- a/spacy/pipeline.pyx +++ b/spacy/pipeline.pyx @@ -1172,7 +1172,7 @@ cdef class DependencyParser(Parser): def init_multitask_objectives(self, get_gold_tuples, pipeline, sgd=None, **cfg): for labeller in self._multitasks: - tok2vec = self.model[0] + tok2vec = self.model.tok2vec labeller.begin_training(get_gold_tuples, pipeline=pipeline, tok2vec=tok2vec, sgd=sgd) @@ -1193,7 +1193,7 @@ cdef class EntityRecognizer(Parser): def init_multitask_objectives(self, get_gold_tuples, pipeline, sgd=None, **cfg): for labeller in self._multitasks: - tok2vec = self.model[0] + tok2vec = self.model.tok2vec labeller.begin_training(get_gold_tuples, pipeline=pipeline, tok2vec=tok2vec)