adelorenzo/stable
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

reformat documentation

Browse Source
This commit is contained in:
Yinon Polak 2023-03-09 13:01:04 +02:00
parent 6f962362f2
commit c9eee2944b
1 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
freqtrade/freqai/base_models/PyTorchModelTrainer.py
View File

@ -55,8 +55,8 @@ class PyTorchModelTrainer:
:param device: The device to use for training (e.g. 'cpu', 'cuda'). :param device: The device to use for training (e.g. 'cpu', 'cuda').
:param batch_size: The size of the batches to use during training. :param batch_size: The size of the batches to use during training.
:param max_iters: The number of training iterations to run. :param max_iters: The number of training iterations to run.
iteration here refers to the number of times we call self.optimizer.step(). iteration here refers to the number of times we call
used to calculate n_epochs. self.optimizer.step(). used to calculate n_epochs.
:param eval_iters: The number of iterations used to estimate the loss. :param eval_iters: The number of iterations used to estimate the loss.
:param init_model: A dictionary containing the initial model/optimizer :param init_model: A dictionary containing the initial model/optimizer
state_dict and model_meta_data saved by self.save() method. state_dict and model_meta_data saved by self.save() method.