pytorch-lightning
仓库创建 2025年11月3日最近提交 1 个月前SkillHot 收录 20 天前
▸ 精选理由
显著减少样板代码并平滑扩展到多卡/多节点，适合快速开发与实验迭代。
这个 Skill 做什么
提供高层 Trainer API，简化 PyTorch 训练代码并无缝扩展到分布式与加速后端。
把 PyTorch 的训练代码变得又短又规整，Trainer 会把训练、验证、checkpoint、callback 等流程抽象好，几行代码就能跑起完整训练。需要跑分布式训练或把单机代码扩展到多机/加速后端（如 DDP/FSDP/DeepSpeed）时特别好用。好处是把样板代码全藏起来，同时保留原生 PyTorch 的灵活性，方便做可复用的训练流水线。
▸ 展开 SKILL.md 英文原文
High-level PyTorch framework with Trainer class, automatic distributed training (DDP/FSDP/DeepSpeed), callbacks system, and minimal boilerplate. Scales from laptop to supercomputer with same code. Use when you want clean training loops with built-in best practices.
1.1w
Stars
817
Forks
40
仓库内 Skill
+704
7 日增星
安装 / 使用
给你的 Agent 一句话（通用）
帮我安装这个 skill：https://raw.githubusercontent.com/Orchestra-Research/AI-Research-SKILLs/main/08-distributed-training/pytorch-lightning/SKILL.md或 curl 直取 SKILL.md
curl -fsSL "https://raw.githubusercontent.com/Orchestra-Research/AI-Research-SKILLs/main/08-distributed-training/pytorch-lightning/SKILL.md"SKILL.MD 节选查看完整文件 ↗
# PyTorch Lightning - High-Level Training Framework
## Quick start
PyTorch Lightning organizes PyTorch code to eliminate boilerplate while maintaining flexibility.
**Installation**:
```bash
pip install lightning
```
**Convert PyTorch to Lightning** (3 steps):
```python
import lightning as L
import torch
from torch import nn
from torch.utils.data import DataLoader, Dataset
# Step 1: Define LightningModule (organize your PyTorch code)
class LitModel(L.LightningModule):
def __init__(self, hidden_size=128):
super().__init__()
self.model = nn.Sequential(
nn.Linear(28 * 28, hidden_size),
nn.ReLU(),
nn.Linear(hidden_size, 10)
)
via SKILL·HOT · 数据来自 GitHub 公开信息 · 原文版权归作者所有