CTYPTO/freqtrade
1
0
Fork 0
mirror of https://github.com/freqtrade/freqtrade.git synced 2026-02-06 14:20:24 +00:00
Code Issues Packages Projects Releases Wiki Activity

pytorch - bugfix - explicitly assign tensor to var as .to() is not inplace operation

Browse Source
This commit is contained in:
yinon
2023-07-13 15:37:50 +00:00
parent 3cf419cbcd
commit 9cb45a3810
1 changed files with 2 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
freqtrade/freqai/torch/PyTorchModelTrainer.py
View File

@@ -83,8 +83,8 @@ class PyTorchModelTrainer(PyTorchTrainerInterface):
for i, batch_data in enumerate(data_loaders_dictionary["train"]):
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)