2.6 KiB
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 Lite Examples
We show how to accelerate your PyTorch code with Lightning Lite with minimal code changes. You stay in full control of the training loop.
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.