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

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

Browse Source
This commit is contained in:
yinon 2023-08-04 12:51:42 +00:00
parent 836d7b885a
commit 777d25192c
1 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

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

@ -113,8 +113,8 @@ class PyTorchModelTrainer(PyTorchTrainerInterface):
self.model.eval() self.model.eval()
for _, batch_data in enumerate(data_loader_dictionary[split]): for _, batch_data in enumerate(data_loader_dictionary[split]):
xb, yb = batch_data xb, yb = batch_data
xb.to(self.device) xb = xb.to(self.device)
yb.to(self.device) yb = yb.to(self.device)
yb_pred = self.model(xb) yb_pred = self.model(xb)
loss = self.criterion(yb_pred, yb) loss = self.criterion(yb_pred, yb)