SwanHubX/SwanLab

해결하는 문제

SwanLab은 모델 학습 팀을 위해 설계된 AI 학습 분석 및 메트릭 관찰 플랫폼입니다. 시각화, 자동 로깅, 하이퍼파라미터 기록 및 실험 비교를 위한 도구를 제공함으로써 복잡한 학습 프로세스를 추적하는 문제를 해결하며, 연구자가 학습 문제를 식별하고 모델 반복을 가속화하도록 돕습니다.

작동 방식

사용자는 가벼운 SDK를 머신러닝 파이프라인에 통합하여 스칼라, 이미지, 오디오, 텍스트, 비디오 및 3D 포인트 클라우드를 로깅합니다. 그런 다음 플랫폼은 중앙 집중식 대시보드를 통해 이 데이터를 시각화합니다. 원격 모니터링을 위한 클라우드 기반 버전과 오프라인 환경을 위한 셀프 호스팅 커뮤니티 버전을 모두 지원합니다. 또한 PyTorch, HuggingFace Transformers, LLaMA Factory와 같은 30개 이상의 주요 프레임워크와 통합됩니다.

대상 사용자

학습 진행 상황을 모니터링하고, 최적의 하이퍼파라미터를 찾기 위해 여러 실험을 비교하며, 팀원과 실시간으로 협업해야 하는 AI 연구자 및 모델 학습 팀을 위해 구축되었습니다.

주요 특징

• 포괄적인 메트릭 추적: 스칼라, 3D 포인트 클라우드, 생화학 분자 및 커스텀 ECharts를 포함한 다양한 데이터 유형을 지원합니다.
• 광범위한 프레임워크 통합: PyTorch Lightning, Ultralytics, XGBoost를 포함한 30개 이상의 프레임워크와 원활하게 통합됩니다.
• 폭넓은 하드웨어 모니터링: NVIDIA GPUs, Ascend NPUs, AMD ROCm 및 기타 다양한 전문 AI 가속기를 포함한 광범위한 하드웨어에 대한 실시간 시스템 수준 모니터링을 제공합니다.
• 협업 도구: 프로젝트 공유, 팀 협업 및 Slack, Discord, Lark를 통한 알림용 플러그인 확장을 제공합니다.
• 유연한 배포: 관리형 클라우드 서비스와 Docker 및 Kubernetes를 통한 셀프 호스팅 옵션을 모두 제공합니다.