mirror of
https://github.com/explosion/spaCy.git
synced 2025-08-04 12:20:20 +03:00
Use NumpyOps() to copy data to CPU
Co-authored-by: Madeesh Kannan <shadeMe@users.noreply.github.com>
This commit is contained in:
parent
95a4835342
commit
361b64e648
|
@ -160,8 +160,7 @@ 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()
|
||||
NumpyOps().asarray(doc_scores)
|
||||
doc_compat_guesses = []
|
||||
for i, token in enumerate(doc):
|
||||
for _ in range(self.top_k):
|
||||
|
|
Loading…
Reference in New Issue
Block a user