mirror of
https://github.com/explosion/spaCy.git
synced 2025-08-04 12:20:20 +03:00
Handle arrays on GPU
This commit is contained in:
parent
278181ab59
commit
9430306076
|
@ -160,6 +160,8 @@ class EditTreeLemmatizer(TrainablePipe):
|
|||
def _scores2guesses(self, docs, scores):
|
||||
guesses = []
|
||||
for doc, doc_scores in zip(docs, scores):
|
||||
if not isinstance(doc_scores, np.ndarray):
|
||||
doc_scores = doc_scores.get()
|
||||
doc_compat_guesses = []
|
||||
for i, token in enumerate(doc):
|
||||
for _ in range(self.top_k):
|
||||
|
|
Loading…
Reference in New Issue
Block a user