c8059d7bfd
* Add 2D parallel example * replace with torchtitan code |
||
---|---|---|
.. | ||
app | ||
fabric | ||
pytorch | ||
README.md | ||
run_fabric_examples.sh | ||
run_pl_examples.sh |
README.md
Examples
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.
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.