ksuit.schedules.functional
==========================

.. py:module:: ksuit.schedules.functional


Functions
---------

.. autoapisummary::

   ksuit.schedules.functional.linear
   ksuit.schedules.functional.cosine
   ksuit.schedules.functional.polynomial


Module Contents
---------------

.. py:function:: linear(step, total_steps)

   linearly increasing from [0 to 1]


.. py:function:: cosine(step, total_steps)

   cosine schedule from [0 to 1]


.. py:function:: polynomial(step, total_steps, power)

   polynomial schedule from [0 to 1]
   https://pytorch.org/docs/stable/generated/torch.optim.lr_scheduler.PolynomialLR.html