From bc9454e0f9a2a67e1b9ffdb6ade4fe500aee7682 Mon Sep 17 00:00:00 2001 From: Yinon Polak Date: Mon, 3 Apr 2023 16:36:38 +0300 Subject: [PATCH] add device to data convertor class doc --- freqtrade/freqai/torch/PyTorchDataConvertor.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/freqtrade/freqai/torch/PyTorchDataConvertor.py b/freqtrade/freqai/torch/PyTorchDataConvertor.py index 1c948c72e..1070b0fb5 100644 --- a/freqtrade/freqai/torch/PyTorchDataConvertor.py +++ b/freqtrade/freqai/torch/PyTorchDataConvertor.py @@ -11,7 +11,7 @@ class PyTorchDataConvertor(ABC): def convert_x(self, df: pd.DataFrame, device: Optional[str] = None) -> Tuple[torch.Tensor, ...]: """ :param df: "*_features" dataframe. - :param device: cpu/gpu. + :param device: The device to use for training (e.g. 'cpu', 'cuda'). :returns: tuple of tensors. """ @@ -19,7 +19,7 @@ class PyTorchDataConvertor(ABC): def convert_y(self, df: pd.DataFrame, device: Optional[str] = None) -> Tuple[torch.Tensor, ...]: """ :param df: "*_labels" dataframe. - :param device: cpu/gpu. + :param device: The device to use for training (e.g. 'cpu', 'cuda'). :returns: tuple of tensors. """