lightning/docs/source-pytorch/levels/advanced_level_17.rst

38 lines
850 B
ReStructuredText

: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>