diff --git a/freqtrade/freqai/base_models/PyTorchModelTrainer.py b/freqtrade/freqai/base_models/PyTorchModelTrainer.py
index 52fb0ceb5..02ff35085 100644
--- a/freqtrade/freqai/base_models/PyTorchModelTrainer.py
+++ b/freqtrade/freqai/base_models/PyTorchModelTrainer.py
@@ -84,7 +84,7 @@ class PyTorchModelTrainer:
                 loss = self.criterion(yb_pred, yb)
                 losses[i] = loss.item()
 
-            loss_dictionary[split] = losses.mean()
+            loss_dictionary[split] = losses.mean().item()
 
         self.model.train()
         return loss_dictionary