lightning/pl_examples

Examples

Our most robust examples showing all sorts of implementations can be found in our sister library lightning-bolts.

---

Basic examples

In this folder we add 3 simple examples:

• MNIST Classifier (defines the model inside the LightningModule).
• Image Classifier (trains arbitrary datasets with arbitrary backbones).
• Autoencoder (shows how the LightningModule can be used as a system)

---

Domain examples

This folder contains older examples. You should instead use the examples in lightning-bolts for advanced use cases.

---

Loop examples

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

• K-fold Cross Validation Loop: Implemenation 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.