torchcell.scheduler
===================

.. module:: torchcell.scheduler

.. currentmodule:: torchcell.scheduler

Learning-rate schedulers. :class:`~torchcell.scheduler.cosine_annealing_warmup.CosineAnnealingWarmupRestarts` is cosine annealing with a linear warmup and cyclic restarts.

.. contents:: Contents
    :local:

``cosine_annealing_warmup``
---------------------------

Cosine annealing learning-rate scheduler with warmup and cyclic restarts.

Classes
~~~~~~~

.. autosummary::
   :nosignatures:
   :toctree: ../generated
   :template: autosummary/class.rst

   cosine_annealing_warmup.CosineAnnealingWarmupRestarts