explosion/spaCy
1
1
Fork 0
mirror of https://github.com/explosion/spaCy.git synced 2025-06-29 17:33:10 +03:00
Code Issues Packages Projects Releases Wiki Activity

Fix support of maxout pieces in parser

Browse Source
This commit is contained in:
Matthew Honnibal 2017-10-20 03:07:17 +02:00
parent a8850b4282
commit 827cd8a883
1 changed files with 4 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
spacy/syntax/nn_parser.pyx
View File

@ -116,8 +116,8 @@ cdef class precompute_hiddens:
else: else:
cached = gpu_cached cached = gpu_cached
self.nF = cached.shape[1] self.nF = cached.shape[1]
self.nO = cached.shape[2]
self.nP = getattr(lower_model, 'nP', 1) self.nP = getattr(lower_model, 'nP', 1)
self.nO = cached.shape[2] // self.nP
self.ops = lower_model.ops self.ops = lower_model.ops
self.bias = lower_model.b self.bias = lower_model.b
self._is_synchronized = False self._is_synchronized = False
@ -174,7 +174,8 @@ cdef class precompute_hiddens:
if self.nP == 1: if self.nP == 1:
return d_best * mask return d_best * mask
else: else:
return self.ops.backprop_maxout(d_best, mask, self.nP) d_vector = self.ops.backprop_maxout(d_best, mask, self.nP)
return d_vector.reshape((d_vector.shape[0], self.nO*self.nP))
return state_vector, backprop_nonlinearity return state_vector, backprop_nonlinearity
@ -267,6 +268,7 @@ cdef class Parser:
tok2vec = chain(tok2vec, flatten) tok2vec = chain(tok2vec, flatten)
lower = PrecomputableAffine(hidden_width * parser_maxout_pieces, lower = PrecomputableAffine(hidden_width * parser_maxout_pieces,
nF=cls.nr_feature, nI=token_vector_width) nF=cls.nr_feature, nI=token_vector_width)
lower.nP = parser_maxout_pieces
with Model.use_device('cpu'): with Model.use_device('cpu'):
upper = chain( upper = chain(