Fix wiring of pre-trained vectors in parser loading

This commit is contained in:
Matthew Honnibal 2017-09-17 05:47:34 -05:00
parent 31c2e91c35
commit 2b0efc77ae
1 changed files with 1 additions and 0 deletions

View File

@ -871,6 +871,7 @@ cdef class Parser:
if self.model is True: if self.model is True:
self.model, cfg = self.Model(self.moves.n_moves, self.model, cfg = self.Model(self.moves.n_moves,
pretrained_dims=self.vocab.vectors_length) pretrained_dims=self.vocab.vectors_length)
cfg['pretrained_dims'] = self.vocab.vectors_length
else: else:
cfg = {} cfg = {}
cfg['pretrained_dims'] = self.vocab.vectors_length cfg['pretrained_dims'] = self.vocab.vectors_length