mohammadian7/freqtrade
1
0
Fork 0
Code Issues Pull Requests Actions 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
Download Patch File Download Diff File Expand all files Collapse all files
freqtrade/freqai/torch/PyTorchModelTrainer.py
+2 -2
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)