Back to Pytorch Lightning

Build a Model

docs/source-pytorch/model/build_model.rst

2.6.41.3 KB
Original Source

:orphan:

############# Build a Model #############

.. raw:: html

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

.. Add callout items below this line

.. displayitem:: :header: 1: Train a model :description: Build a model to learn the basic ideas of Lightning :col_css: col-md-4 :button_link: train_model_basic.html :height: 150 :tag: basic

.. displayitem:: :header: 2: Validate and test a model :description: Add a validation and test data split to avoid overfitting. :col_css: col-md-4 :button_link: ../common/evaluation_basic.html :height: 150 :tag: basic

.. displayitem:: :header: 3: Supercharge training :description: Enable state-of-the-art training techniques with the Trainer features. :col_css: col-md-4 :button_link: build_model_intermediate.html :height: 150 :tag: intermediate

.. displayitem:: :header: LightningModule API :description: Dig into LightningModule API in depth :col_css: col-md-4 :button_link: ../common/lightning_module.html#lightningmodule-api :height: 150

.. displayitem:: :header: Trainer API :description: Dig into Trainer API in depth :col_css: col-md-4 :button_link: ../common/trainer.html#trainer-class-api :height: 150

.. raw:: html

    </div>
</div>