From 84fca256005ddddddc0dbbb28a0b0a7050169dc9 Mon Sep 17 00:00:00 2001 From: Lucas Cazzonelli Date: Wed, 30 Aug 2023 13:15:33 +0200 Subject: [PATCH] fix: tensor conversion with gpu The conversion of GPU tensors to numpy arrays caused an error, which should now be fixed. --- deep_river/utils/tensor_conversion.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/deep_river/utils/tensor_conversion.py b/deep_river/utils/tensor_conversion.py index 472f856..2464bc7 100644 --- a/deep_river/utils/tensor_conversion.py +++ b/deep_river/utils/tensor_conversion.py @@ -156,7 +156,7 @@ def output2proba( else: preds = torch.sigmoid(preds) - preds_np = preds.detach().numpy() + preds_np = preds.numpy(force=True) if preds_np.shape[1] == 1: preds_np = np.hstack((preds_np, 1 - preds_np)) n_unobserved_classes = preds_np.shape[1] - len(classes)