fix rendering part 1

This commit is contained in:
giovanni
2025-03-13 22:31:26 +01:00
committed by FilippoOlivo
parent 5e6aa61592
commit 3d842cb9ec
15 changed files with 89 additions and 63 deletions

View File

@@ -21,8 +21,8 @@ class TorchScheduler(Scheduler):
"""
Initialization of the :class:`TorchScheduler` class.
:param torch.optim.LRScheduler scheduler_class: The PyTorch scheduler
class.
:param torch.optim.LRScheduler scheduler_class: A
:class:`torch.optim.LRScheduler` class.
:param dict kwargs: Additional parameters passed to `scheduler_class`,
see more: <https://pytorch.org/docs/stable/optim.html#algorithms>_.
"""