Watch out for the (PyTorch) Lightning

Table of Contents

1. What is PyTorch Lightning
2. Installing and Getting Started
3. Learning with Lighting
4. Grid AI
5. Helpful Links

What is PyTorch Lightning

Turn PyTorch code into Lightning code.

PyTorch Code of model:

Lightning Code of model:

PyTorch Code of training:

Lightning Code of training:

More about PyTorch here:

Introduction to PyTorch

Tina Huynh ・ May 2 ・ 3 min read

#python #machinelearning #deeplearning #beginners

Back to TOC

Installing and Getting Started

Install PyTorch Lightning by running:

pip install pytorch-lightning

or

conda install pytorch-lightning -c conda-forge

Lighting in 15 Minutes

PyTorch Lighting's documentation has skill levels you are able to walk through, each teaching you a new skill.
Level 1: Train a model
Level 2: Add a validation and test set
Level 3: Use pretrained models
Level 4: Enable script parameters
.
.
Level 10: Understand your model
Level 11: Explore SOTA scaling techniques
.
.
Level 20: Explore HPUs
Level 21: Master TPUs
Level 22: Reach 1 trillion parameters on GPUs
.
.
.
Level 26: Make your own profiler

Learning with Lighting

There are many tutorials and other resources on PyTorch Lighting's website you can utilize to start your journey with Lighting.

Tutorials

Glossary

Common WorkFlows

Slack Community

Join their Slack to ask the community questions and check out the GitHub here.

Grid AI

From the creators of PyTorch Lighting, you can get started with grid and train countless of ML models from your laptop with zero code changes!

Introducing Grid AI

Helpful Links

• PyTorch Lightning MasterClass
• Lightning Research Talks
• Getting Started with PyTorch Lightning
• PyTorch Ecosystem
• PyTorch Lightning for Dummies
• Training Neural Networks using PyTorch Lightning

Happy coding!