pytorch-lightning - PyTorch Lightning Training Framework
High-level PyTorch framework with automatic distributed training and minimal boilerplate
Tags
Updated: 2026-05-28Capabilities
Typical Inputs
Typical Outputs
What this skill does
- train model
- validate model
- test model
- run distributed training
- log metrics
- save checkpoint
- configure optimizer
- schedule learning rate
- monitor learning rate
- stop early
- show progress
- organize code
- accumulate gradients
- use mixed precision
- load checkpoint
Inputs
- PyTorch model
- training dataset
- validation dataset
- test dataset
- PyTorch DataLoader
Outputs
- trained model
- validation metrics
- test metrics
- checkpoint file
- training logs
Requirements
- PyTorch
- Lightning library
- GPU or CPU
- Python environment