mohammadian7/freqtrade
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Update PyTorchModelTrainer.py

Browse Source 
The n_epochs should be defined using the `max` not the `min` function.
This commit is contained in:
Robert Caulk
2024-03-25 09:21:32 +01:00
committed by GitHub
parent 526d7fad62
commit 18e34632d8
1 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files
freqtrade/freqai/torch/PyTorchModelTrainer.py
+1 -1
View File
@@ -152,7 +152,7 @@ class PyTorchModelTrainer(PyTorchTrainerInterface):
""" """
assert isinstance(self.n_steps, int), "Either `n_steps` or `n_epochs` should be set." assert isinstance(self.n_steps, int), "Either `n_steps` or `n_epochs` should be set."
n_batches = n_obs // self.batch_size n_batches = n_obs // self.batch_size
n_epochs = min(self.n_steps // n_batches, 1) n_epochs = max(self.n_steps // n_batches, 1)
if n_epochs <= 10: if n_epochs <= 10:
logger.warning( logger.warning(
f"Setting low n_epochs: {n_epochs}. " f"Setting low n_epochs: {n_epochs}. "