lychiyu/freqtrade_origin
1
0
Fork 0
mirror of https://github.com/freqtrade/freqtrade.git synced 2024-11-14 04:03:55 +00:00
Code Issues Actions 1 Packages Projects Releases Wiki Activity

Merge pull request #10001 from freqtrade/fix/epoch-definition

Browse Source 
Update PyTorchModelTrainer.py
This commit is contained in:
Matthias 2024-03-25 19:50:02 +01:00 committed by GitHub
commit 41d508867e
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
freqtrade/freqai/torch/PyTorchModelTrainer.py
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}. "