메인 콘텐츠로 건너뛰기
W&B Python SDK는 wandb를 통해 사용 가능하며, 모델을 트레이닝하고 파인튜닝할 수 있을 뿐만 아니라, 실험 단계부터 프로덕션까지 모델을 관리할 수 있도록 합니다.
이 SDK로 트레이닝 및 파인튜닝 작업을 수행한 후에는 Public API를 사용하여 로깅된 데이터를 조회하고 분석할 수 있으며, Reports and Workspaces API를 사용하여 작업 내용을 요약한 웹 게시용 리포트를 생성할 수 있습니다.

설치 및 설정

가입하고 API 키 만들기

W&B에서 머신을 인증하려면 먼저 User Settings에서 API 키를 생성해야 합니다.

패키지 설치 및 임포트

W&B 라이브러리를 설치합니다.
pip install wandb

W&B Python SDK 임포트하기:

import wandb

# 팀 entity를 지정하세요
entity = "<team_entity>"

# run이 기록될 프로젝트
project = "my-awesome-project"

with wandb.init(entity=entity, project=project) as run:
   run.log({"accuracy": 0.9, "loss": 0.1})