pytorch-lightning · davila7 — скилл для Claude Code: что делает и как установить
Помогает Claude Code структурировать код PyTorch в модули LightningModule и настраивать обучение на нескольких GPU или TPU. Убирает рутину вокруг циклов обучения и пайплайнов данных.
npx claude-code-templates@latest --skill scientific/pytorch-lightningЧто делает pytorch-lightning
Скилл снабжает Claude Code инструкциями и шаблонами для переноса кода на PyTorch Lightning. Логика нейросети упаковывается в методы LightningModule, а пайплайн данных изолируется внутри LightningDataModule. В состав скилла входят готовые скрипты template_lightning_module.py и template_datamodule.py.
Для запуска обучения скилл помогает настроить Trainer: включить смешанную точность или задать отсечение градиентов. Агент умеет подключать логгеры вроде TensorBoard и WandbLogger, а также колбэки ModelCheckpoint и EarlyStopping. Проверить пайплайн на одном батче можно через fast_dev_run=True.
В распределённом обучении скилл чётко разделяет стратегии: DDP предлагается для моделей до 500M параметров, а FSDP и DeepSpeed выбираются для более крупных сетей. Без пакетов lightning и torch в окружении сгенерированный код не запустится.
Какие задачи решает
- переписать чистый скрипт PyTorch в структуру
LightningModule - изолировать подготовку датасетов в
LightningDataModule - сконфигурировать
Trainerдля распределённого обучения на нескольких GPU через DDP или FSDP - добавить сохранение лучших чекпоинтов и раннюю остановку через колбэки
Примеры запросов агенту
- Оберни модель из
model.pyв LightningModule и добавь оптимизатор Adam - Создай класс LightningDataModule для загрузки датасета с размером батча 32
- Настрой Trainer для распределённого обучения на 4 GPU с использованием FSDP
Как установить pytorch-lightning
1Через claude-code-templates
npx claude-code-templates@latest --skill scientific/pytorch-lightning2Через Skills CLI (Vercel)
npx skills add davila7/claude-code-templates --skill pytorch-lightning -a claude-code3Вручную: положить файлы
~/.claude/skills/pytorch-lightning/SKILL.mdПапка скилла целиком; для одного проекта — .claude/skills/pytorch-lightning/
Вручную: положить файлы
.agents/skills/pytorch-lightning/SKILL.mdДля всех проектов — ~/.agents/skills/
Вручную: положить файлы
.cursor/skills/pytorch-lightning/SKILL.mdCursor читает и .agents/skills/, и .claude/skills/
1GitHub CLI
gh skill install davila7/claude-code-templates pytorch-lightning2Вручную: положить файлы
.github/skills/pytorch-lightning/SKILL.mdКомандой платформы
gemini skills install https://github.com/davila7/claude-code-templates --path cli-tool/components/skills/scientific/pytorch-lightningВручную: положить файлы
.devin/skills/pytorch-lightning/SKILL.mdСтарый путь .windsurf/skills/ тоже читается
Claude.ai
Упакуйте папку в zip и загрузите в Settings → Capabilities → Skills
По шагам, с командами для каждой платформы: Как установить скилл в Claude Code: пять способов
Вопросы о pytorch-lightning
Как установить pytorch-lightning в Claude Code?
Выполните команду npx claude-code-templates@latest --skill scientific/pytorch-lightning. Также можно подключить скилл командой npx skills add davila7/claude-code-templates --skill pytorch-lightning -a claude-code.
Что делает скилл pytorch-lightning?
Он даёт агенту архитектурные правила и готовые шаблоны для перевода нейросетей на PyTorch Lightning. С ним Claude Code быстрее формирует модули моделей, пайплайны данных и конфигурации тренера.
Какую стратегию распределённого обучения предлагает скилл?
Для моделей размером менее 500M параметров выбирается стратегия DDP. Для более крупных сетей от 500M параметров скилл рекомендует использовать FSDP или DeepSpeed.
Похожие скиллы
Все скиллы по теме «ИИ и ML» →Сначала — pytorch-lightning от других авторов.
