diff --git a/freqtrade/freqai/torch/PyTorchModelTrainer.py b/freqtrade/freqai/torch/PyTorchModelTrainer.py
index dc34e8907..e7c4d53be 100644
--- a/freqtrade/freqai/torch/PyTorchModelTrainer.py
+++ b/freqtrade/freqai/torch/PyTorchModelTrainer.py
@@ -113,8 +113,8 @@ class PyTorchModelTrainer(PyTorchTrainerInterface):
         self.model.eval()
         for _, batch_data in enumerate(data_loader_dictionary[split]):
             xb, yb = batch_data
-            xb.to(self.device)
-            yb.to(self.device)
+            xb = xb.to(self.device)
+            yb = yb.to(self.device)
 
             yb_pred = self.model(xb)
             loss = self.criterion(yb_pred, yb)