CTYPTO/freqtrade
1
0
Fork 0
mirror of https://github.com/freqtrade/freqtrade.git synced 2025-11-29 00:23:07 +00:00
Code Issues 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
parent 526d7fad62 18e34632d8
commit 41d508867e
1 changed files with 1 additions and 1 deletions
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."
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:
logger.warning(
f"Setting low n_epochs: {n_epochs}. "