pytorch-lightning
Cleaner PyTorch training code that scales to more GPUs.
What it does
Guides the agent in organizing PyTorch training code into Lightning modules with managed training, validation, and distributed execution.
When to use it
Use when developing organized training loops with built-in callbacks and logging, or switching between single-GPU, multi-GPU, and TPU training.
How to use it
It changes how the agent organizes PyTorch training code, using Lightning modules and Trainer-based workflows.
What you provide
- An existing project
What you need · 3
Uses Python to run the training script; no Python version requirement is given.
The training examples use the lightning package, and the quick start gives its installation command.
The core examples import torch for neural-network layers, data loading, and optimization.
Access · 1
This skill
lightning
Installs the lightning package