.. role:: hidden
    :class: hidden-section
.. currentmodule:: torchtune.training


get_cosine_schedule_with_warmup
===============================

.. autofunction:: get_cosine_schedule_with_warmup