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

pytorch - trainer - set default usage of n_epochs instead of max_iters

Browse Source
This commit is contained in:
yinon
2023-08-04 12:50:01 +00:00
parent 8ebfb731d8
commit 836d7b885a
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

@@ -50,8 +50,8 @@ class PyTorchModelTrainer(PyTorchTrainerInterface):
self.criterion = criterion
self.model_meta_data = model_meta_data
self.device = device
self.max_iters: int = kwargs.get("max_iters", 100)
self.n_epochs: Optional[int] = kwargs.get("n_epochs", None)
self.max_iters: int = kwargs.get("max_iters", None)
self.n_epochs: Optional[int] = kwargs.get("n_epochs", 10)
if not self.max_iters and not self.n_epochs:
raise Exception("Either `max_iters` or `n_epochs` should be set.")