From 5dd60eda3693df999f80305cb60f2c6fc3b22a3d Mon Sep 17 00:00:00 2001 From: Yinon Polak Date: Mon, 6 Mar 2023 19:37:08 +0200 Subject: [PATCH] type hints fixes --- freqtrade/freqai/base_models/PyTorchModelTrainer.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) 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