type hints fixes

2023-03-06 19:37:08 +02:00
parent 8acdd0b47c
commit 5dd60eda36
1 changed files with 1 additions and 1 deletions
--- 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