2.0 KiB
2.0 KiB
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 several starter examples:
- MNIST Classifier: Shows how to define the model inside the
LightningModule. - Image Classifier: Trains arbitrary datasets with arbitrary backbones.
- Autoencoder: Shows how the
LightningModulecan be used as a system. - Profiler: Shows the basic usage of the PyTorch profilers and how to inspect traces in Google Chrome.
- Image Classifier with DALI: Shows how to use NVIDIA DALI with Lightning.
- Mnist Datamodule: Shows how to define a simple
LightningDataModuleusing the MNIST dataset.
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.