lychiyu/freqtrade_origin
1
0
Fork 0
mirror of https://github.com/freqtrade/freqtrade.git synced 2024-11-10 10:21:59 +00:00
Code Issues Actions 2 Packages Projects Releases Wiki Activity

type hints fixes

Browse Source
This commit is contained in:
Yinon Polak 2023-03-06 19:37:08 +02:00
parent 8acdd0b47c
commit 5dd60eda36
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
freqtrade/freqai/base_models/PyTorchModelTrainer.py
View File

@ -84,7 +84,7 @@ class PyTorchModelTrainer:
loss = self.criterion(yb_pred, yb) loss = self.criterion(yb_pred, yb)
losses[i] = loss.item() losses[i] = loss.item()
loss_dictionary[split] = losses.mean() loss_dictionary[split] = losses.mean().item()
self.model.train() self.model.train()
return loss_dictionary return loss_dictionary