explosion/spaCy
1
1
Fork 0
mirror of https://github.com/explosion/spaCy.git synced 2025-02-22 22:40:32 +03:00
Code Issues Packages Projects Releases Wiki Activity

Fix depth=0, maxout pieces=0

Browse Source
This commit is contained in:
Matthw Honnibal 2019-11-19 11:17:56 +01:00
parent 511ced11ff
commit 31af39f851
1 changed files with 4 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
spacy/syntax/_parser_model.pyx
View File

@ -493,13 +493,16 @@ cdef class precompute_hiddens:
if self.activation == "relu": if self.activation == "relu":
mask = state_vector >= 0. mask = state_vector >= 0.
state_vector *= mask state_vector *= mask
else:
mask = None
def backprop_nonlinearity(d_best, sgd=None): def backprop_nonlinearity(d_best, sgd=None):
if isinstance(d_best, numpy.ndarray): if isinstance(d_best, numpy.ndarray):
ops = NumpyOps() ops = NumpyOps()
else: else:
ops = CupyOps() ops = CupyOps()
mask_ = ops.asarray(mask) if mask is not None:
mask_ = ops.asarray(mask)
# This will usually be on GPU # This will usually be on GPU
d_best = ops.asarray(d_best) d_best = ops.asarray(d_best)