explosion/spaCy
1
1
Fork 0
mirror of https://github.com/explosion/spaCy.git synced 2024-12-26 18:06:29 +03:00
Code Issues Packages Projects Releases Wiki Activity

Fix tensorizer on GPU

Browse Source
This commit is contained in:
Matthew Honnibal 2017-11-05 15:34:40 +01:00
parent 6e5181bbaa
commit 2b35bb76ad
2 changed files with 10 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
spacy/pipeline.pyx
View File

@ -415,7 +415,11 @@ class Tagger(Pipe):
vocab.morphology.assign_tag_id(&doc.c[j], tag_id)
idx += 1
if tensors is not None:
doc.extend_tensor(tensors[i])
if isinstance(doc.tensor, numpy.ndarray) \
and not isinstance(tensors[i], numpy.ndarray):
doc.extend_tensor(tensors[i].get())
else:
doc.extend_tensor(tensors[i])
doc.is_tagged = True
def update(self, docs, golds, drop=0., sgd=None, losses=None):

6
spacy/syntax/nn_parser.pyx
View File

@ -751,7 +751,11 @@ cdef class Parser:
for j in range(doc.length):
doc.c[j] = state.c._sent[j]
if tensors is not None:
doc.extend_tensor(tensors[i])
if isinstance(doc.tensor, numpy.ndarray) \
and not isinstance(tensors[i], numpy.ndarray):
doc.extend_tensor(tensors[i].get())
else:
doc.extend_tensor(tensors[i])
self.moves.finalize_doc(doc)
for hook in self.postprocesses: