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

pytorch - trainer - revert load changes

Browse Source
This commit is contained in:
yinon
2023-08-04 12:52:55 +00:00
parent 777d25192c
commit d17bf6350d
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

@@ -181,8 +181,8 @@ class PyTorchModelTrainer(PyTorchTrainerInterface):
"pytrainer": self
}, path)
def load(self, path: Path, device: Optional[str] = None):
checkpoint = torch.load(path, map_location=device)
def load(self, path: Path):
checkpoint = torch.load(path)
return self.load_from_checkpoint(checkpoint)
def load_from_checkpoint(self, checkpoint: Dict):