Fix wiring of pre-trained vectors in parser loading

2017-09-17 05:47:34 -05:00 · 2017-09-17 05:47:34 -05:00 · 2b0efc77ae
parent 31c2e91c35
commit 2b0efc77ae
1 changed files with 1 additions and 0 deletions
--- a/spacy/syntax/nn_parser.pyx
+++ b/spacy/syntax/nn_parser.pyx
@ -871,6 +871,7 @@ cdef class Parser:
            if self.model is True:
                self.model, cfg = self.Model(self.moves.n_moves,
                                    pretrained_dims=self.vocab.vectors_length)
+                cfg['pretrained_dims'] = self.vocab.vectors_length
            else:
                cfg = {}
            cfg['pretrained_dims'] = self.vocab.vectors_length