Back to Pytorch Lightning

Level 16: Own the training loop

docs/source-pytorch/levels/advanced_level_17.rst

2.6.4849 B
Original Source

:orphan:

############################### Level 16: Own the training loop ###############################

Learn all the ways of owning your raw PyTorch loops with Lightning.


.. raw:: html

<div class="display-card-container">
    <div class="row">

.. Add callout items below this line

.. displayitem:: :header: Enable manual optimization :description: Gain control of the training loop with manual optimization and LightningModule methods. :col_css: col-md-4 :button_link: ../model/build_model_advanced.html :height: 150 :tag: advanced

.. displayitem:: :header: Use Lightning Fabric :description: Write your own training loops using lightning Fabric. :col_css: col-md-4 :button_link: https://lightning.ai/docs/fabric/stable/ :height: 150 :tag: advanced

.. raw:: html

    </div>
</div>