pytorch-lightning - PyTorch Lightning High-Level Training Framework
High-level PyTorch framework with automatic distributed training and callbacks
Tags
Updated: 2026-05-28Capabilities
Typical Inputs
Typical Outputs
What this skill does
- organize code
- train model
- validate model
- test model
- log metrics
- save checkpoint
- monitor training
- configure optimizer
- distribute training
- schedule learning rate
Inputs
- training data
- validation data
- test data
- PyTorch model
- Lightning module
- optimizer configuration
- scheduler configuration
Outputs
- trained model
- validation metrics
- test metrics
- model checkpoints
- training logs
- progress updates
Requirements
- Python environment
- PyTorch
- Lightning library
- GPU/TPU/CPU