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


log_memory_stats
================

.. autofunction:: log_memory_stats