PyTorch Lightning 학습 도우미
PyTorch 코드를 LightningModule 구조로 정리하고 멀티 GPU 학습·로깅·체크포인트까지 설계해 주는 스킬입니다.
개발·코딩중급★ 33,030⑂ 3,248AI 점수 8/10마지막 업데이트: 2026. 8. 9.
무엇을 해주나
복잡한 PyTorch 학습 루프를 PyTorch Lightning 규격에 맞게 정리해 줍니다. LightningModule(학습/검증/테스트/예측 단계), LightningDataModule(데이터 파이프라인), Trainer(멀티 GPU·TPU, 혼합정밀도, 그래디언트 누적) 구성을 자동으로 잡아주고, ModelCheckpoint·EarlyStopping 같은 콜백과 W&B·TensorBoard·MLflow 로깅 연동, DDP/FSDP/DeepSpeed 분산 전략 선택까지 안내합니다. 저장소에는 바로 복사해 쓸 수 있는 템플릿 스크립트와 컴포넌트별 상세 레퍼런스 문서가 함께 들어 있습니다.
이런 분께 추천
- 학습 스크립트가 길어지고 GPU 관리 코드가 뒤엉킨 딥러닝 연구자·엔지니어
- 단일 GPU 실험을 멀티 GPU/노드로 확장하려는 분
- 실험 재현성과 로깅 체계를 잡고 싶은 ML 팀
- 논문 코드나 프로토타입을 정식 프로젝트 구조로 리팩터링하려는 분
활용 예시
- "기존 train.py의 for-loop 학습 코드를 LightningModule로 리팩터링해줘" → 6개 섹션 구조로 정리된 모듈과
configure_optimizers()까지 생성 - "7B 파라미터 모델을 4장 GPU로 학습하려면 어떤 전략이 좋아?" → FSDP 기반 Trainer 설정과 체크포인트 저장 방식 제안
- "W&B 로깅과 EarlyStopping, 최고 성능 모델 저장 콜백을 붙여줘" → 로거·콜백 조합이 적용된 Trainer 코드 작성
· · · 설치 가이드 · · ·
Claude 앱에 설치 (터미널 필요 없음)
- 아래 버튼으로 ZIP 파일을 받으세요.
- Claude 설정 → Capabilities에서 '코드 실행 및 파일 생성'을 켭니다. (한 번만)
- Claude에서 Customize → Skills → + → '스킬 업로드'를 누르고 받은 ZIP을 올립니다.
Claude Code에 설치
Claude에게 맡기기 — 아래 문장을 Claude Code에 붙여넣으세요
클로드스킬마트에서 찾은 스킬을 설치해줘. GitHub 저장소 K-Dense-AI/scientific-agent-skills 의 skills/pytorch-lightning 폴더를 내 ~/.claude/skills/pytorch-lightning/ 에 그대로 복사해줘. 설치가 끝나면 이 스킬로 무엇을 할 수 있는지 한 줄로 알려줘.
직접 명령으로 설치하기
git clone https://github.com/K-Dense-AI/scientific-agent-skills.git /tmp/sas && mkdir -p ~/.claude/skills && cp -r /tmp/sas/skills/pytorch-lightning ~/.claude/skills/⚠ 제3자가 만든 스킬입니다. 설치 전 원본 저장소를 한 번 확인하세요.
- 터미널을 엽니다.
- 저장소를 내려받습니다:
git clone https://github.com/K-Dense-AI/scientific-agent-skills.git - 스킬 폴더를 만듭니다:
mkdir -p ~/.claude/skills - 이 스킬만 복사합니다:
cp -r scientific-agent-skills/skills/pytorch-lightning ~/.claude/skills/ - 파이썬 패키지를 설치합니다:
pip install lightning(GPU 학습 시 CUDA 버전에 맞는 PyTorch 먼저 설치) - Claude Code를 다시 실행한 뒤 "PyTorch Lightning으로 학습 코드 만들어줘"라고 요청하면 스킬이 자동으로 호출됩니다.
GitHub에서 원본 보기 ↗라이선스: MIT