How To Get Started Training With Lightning
Lightning is a lightweight PyTorch wrapper for high-performance AI research, that reduces the boilerplate without limiting flexibility. In this series, we are covering all the tricks Lightning offers to supercharge your machine learning training.
In this video, we give a short intro to the Lightning framework and its main API- the LightningModule. Continue watching to learn more about how to control your training and validation loop, how to scale your model to run on TPUs or multi-node GPUs without changing your code, and more!
More examples of Lightning models can be found in our sister library, Lightning Bolts:
https://github.com/Lightning-AI/lightning-bolts/tree/master/pl_bolts/models