fynance.models.neural_network.BaseNeuralNet.set_lr_scheduler¶
-
BaseNeuralNet.
set_lr_scheduler
(lr_scheduler, **kwargs)¶ Set dynamic learning rate.
Parameters: - lr_scheduler : torch.optim.lr_scheduler._LRScheduler
Method from
torch.optim.lr_scheduler
to wrapself.optimizer
, cf moduletorch.optim.lr_scheduler
in PyTorch documentation [2].- **kwargs
Keyword arguments to pass to the learning rate scheduler.
References
[2] https://pytorch.org/docs/stable/optim.html#how-to-adjust-learning-rate