From 908a2e817a34efaa3d712a99b5be77284336bdef Mon Sep 17 00:00:00 2001 From: Tommaso Falchi Date: Fri, 5 May 2023 15:43:48 +0200 Subject: [PATCH] Align BasePyTorchRegressor tensors to cpu as in BasePyTorchClassifier --- freqtrade/freqai/base_models/BasePyTorchRegressor.py | 3 +-- 1 file changed, 1 insertion(+), 2 deletions(-) diff --git a/freqtrade/freqai/base_models/BasePyTorchRegressor.py b/freqtrade/freqai/base_models/BasePyTorchRegressor.py index ea6fabe49..069387011 100644 --- a/freqtrade/freqai/base_models/BasePyTorchRegressor.py +++ b/freqtrade/freqai/base_models/BasePyTorchRegressor.py @@ -45,6 +45,5 @@ class BasePyTorchRegressor(BasePyTorchModel): device=self.device ) y = self.model.model(x) - y = y.cpu() - pred_df = DataFrame(y.detach().numpy(), columns=[dk.label_list[0]]) + pred_df = DataFrame(y.detach().tolist(), columns=[dk.label_list[0]]) return (pred_df, dk.do_predict)