mohammadian7/freqtrade
1
0
Fork 0
Code Issues Pull Requests Actions 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
freqtrade/freqai/torch/PyTorchModelTrainer.py
+2 -2
View File
@@ -50,8 +50,8 @@ class PyTorchModelTrainer(PyTorchTrainerInterface):
self.criterion = criterion self.criterion = criterion
self.model_meta_data = model_meta_data self.model_meta_data = model_meta_data
self.device = device self.device = device
self.max_iters: int = kwargs.get("max_iters", 100) self.max_iters: int = kwargs.get("max_iters", None)
self.n_epochs: Optional[int] = kwargs.get("n_epochs", None) self.n_epochs: Optional[int] = kwargs.get("n_epochs", 10)
if not self.max_iters and not self.n_epochs: if not self.max_iters and not self.n_epochs:
raise Exception("Either `max_iters` or `n_epochs` should be set.") raise Exception("Either `max_iters` or `n_epochs` should be set.")