explosion/spaCy
1
1
Fork 0
mirror of https://github.com/explosion/spaCy.git synced 2025-08-04 12:20:20 +03:00
Code Issues Packages Projects Releases Wiki Activity

Handle arrays on GPU

Browse Source
This commit is contained in:
richardpaulhudson 2022-12-22 15:00:42 +01:00
parent 278181ab59
commit 9430306076
1 changed files with 2 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
spacy/pipeline/edit_tree_lemmatizer.py
View File

@ -160,6 +160,8 @@ class EditTreeLemmatizer(TrainablePipe):
def _scores2guesses(self, docs, scores): def _scores2guesses(self, docs, scores):
guesses = [] guesses = []
for doc, doc_scores in zip(docs, scores): for doc, doc_scores in zip(docs, scores):
if not isinstance(doc_scores, np.ndarray):
doc_scores = doc_scores.get()
doc_compat_guesses = [] doc_compat_guesses = []
for i, token in enumerate(doc): for i, token in enumerate(doc):
for _ in range(self.top_k): for _ in range(self.top_k):