Commit 6c20a98a authored by Łukasz Kopociński's avatar Łukasz Kopociński

Copy to cpu

parent f02ca270
......@@ -54,4 +54,4 @@ class Predictor:
def predict(self, indices_context: List[Tuple]):
orths, vectors = self._make_vectors(indices_context)
predictions = self._predict(vectors)
return orths, predictions.numpy()
return orths, predictions.cpu().numpy()
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment