Rooba
/
spaCy
mirror of https://github.com/explosion/spaCy.git
1
1
Fork 0
Code Issues Packages Projects Releases Wiki Activity

CPU/GPU compat

This commit is contained in:
Matthew Honnibal 2017-05-07 04:01:11 +02:00
parent f99f5b75dc
commit 700979fb3c
1 changed files with 3 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
spacy/syntax/parser.pyx
View File

@ -256,7 +256,7 @@ cdef class Parser:
self._cost_batch(costs, is_valid, states, golds) self._cost_batch(costs, is_valid, states, golds)
self._set_gradient(d_scores, scores, is_valid, costs) self._set_gradient(d_scores, scores, is_valid, costs)
losses.append(numpy.abs(d_scores).sum()) losses.append(self.model.ops.xp.abs(d_scores).sum())
if force_gold: if force_gold:
softmaxed *= costs <= 0 softmaxed *= costs <= 0
return finish_update(d_scores, sgd=sgd) return finish_update(d_scores, sgd=sgd)
@ -312,9 +312,9 @@ cdef class Parser:
n = gradients.shape[0] n = gradients.shape[0]
scores = scores * is_valid scores = scores * is_valid
g_scores = scores * is_valid * (costs <= 0.) g_scores = scores * is_valid * (costs <= 0.)
exps = numpy.exp(scores - scores.max(axis=1).reshape((n, 1))) exps = self.model.ops.xp.exp(scores - scores.max(axis=1).reshape((n, 1)))
exps *= is_valid exps *= is_valid
g_exps = numpy.exp(g_scores - g_scores.max(axis=1).reshape((n, 1))) g_exps = self.model.ops.xp.exp(g_scores - g_scores.max(axis=1).reshape((n, 1)))
g_exps *= costs <= 0. g_exps *= costs <= 0.
g_exps *= is_valid g_exps *= is_valid
gradients[:] = exps / exps.sum(axis=1).reshape((n, 1)) gradients[:] = exps / exps.sum(axis=1).reshape((n, 1))