Back to Pytorch Lightning

Use a pure PyTorch training loop

docs/source-pytorch/model/own_your_loop.rst

2.6.4532 B
Original Source

:orphan:

################################ Use a pure PyTorch training loop ################################

.. 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: build_model_advanced.html :height: 150 :tag: advanced

.. raw:: html

    </div>
</div>