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
- switch device
- run distributed training
- log metrics
- save checkpoint
- validate model
- test model
- monitor training
- schedule learning rate
- accumulate gradients
Inputs
- PyTorch model
- training data
- validation data
- test data
- GPU devices
Outputs
- trained model
- model checkpoints
- training logs
- validation metrics
- test results
Requirements
- Python
- PyTorch
- Lightning package
- GPU or TPU