pytorch-lightning - PyTorch Lightning 训练框架
高级 PyTorch 框架，支持自动分布式训练和最少样板代码
标签
更新于: 2026-05-28能力
典型输入
典型输出
该技能可以做什么
- 训练模型
- 验证模型
- 测试模型
- 运行分布式训练
- 记录指标
- 保存检查点
- 配置优化器
- 调度学习率
- 监控学习率
- 早停训练
- 显示进度
- 组织代码
- 累积梯度
- 使用混合精度
- 加载检查点
输入
- PyTorch 模型
- 训练数据集
- 验证数据集
- 测试数据集
- PyTorch 数据加载器
输出
- 训练好的模型
- 验证指标
- 测试指标
- 检查点文件
- 训练日志
要求
- PyTorch
- Lightning 库
- GPU 或 CPU
- Python 环境