lightning/examples

Examples

Our most robust examples showing all sorts of implementations can be found in our sister library Lightning Bolts.

---

Note that some examples may rely on new features that are only available in the development branch and may be incompatible with any releases. If you see any errors, you might want to consider switching to a version tag you would like to run examples with. For example, if you're using pytorch-lightning==1.6.4 in your environment and seeing issues, run examples of the tag 1.6.4.

---

Lightning Fabric Examples

We show how to accelerate your PyTorch code with Lightning Fabric with minimal code changes. You stay in full control of the training loop.

• MNIST with vanilla PyTorch
• MNIST with Lightning Fabric

---

Lightning Trainer Examples

In this folder, we have 2 simple examples that showcase the power of the Lightning Trainer.

• Image Classifier (trains arbitrary datasets with arbitrary backbones).
• Autoencoder

---

Domain Examples

This folder contains older examples. You should instead use the examples in Lightning Bolts for advanced use cases.

---

Basic Examples

In this folder, we have 1 simple example:

• Image Classifier + DALI (defines the model inside the LightningModule).

---

Loop examples

Contains implementations leveraging loop customization to enhance the Trainer with new optimization routines.

• K-fold Cross Validation Loop: Implementation of cross validation in a loop and special datamodule.
• Yield Loop: Enables yielding from the training_step like in a Python generator. Useful for automatic optimization with multiple optimizers.