pytorch-lightning - PyTorch Lightning 训练框架
用于分布式训练的高级 PyTorch 框架，支持 DDP/FSDP/DeepSpeed
标签
更新于: 2026-05-28能力
典型输入
典型输出
该技能可以做什么
- 训练模型
- 验证模型
- 测试模型
- 记录指标
- 保存检查点
- 调整学习率
- 配置优化器
- 分布式训练
- 切换设备
- 监控训练
输入
- PyTorch 模型
- DataLoader
- 训练配置
输出
- 训练好的模型
- 模型检查点
- 训练日志
- 训练指标
要求
- lightning 库
- torch
- transformers
- CPU/GPU/TPU/MPS 支持
- FP32/FP16/BF16/FP8 精度