Weights & Biases 사용법

wandb는 Weights&Biases의 줄임말로 모델의 결과를 관리하고 시각화하는 데 사용된다.
주로 원격 환경에서 모델 훈련과 실험을 확인하는 데 사용된다.

wandb 설치 및 로그인

pip install wandb
wandb login

초기 설정

import wandb
wandb.init(
    project="example"
    config={
    "learning_rate": learning_rate,
    "architecture": "CNN",
    "dataset": "Sample",
    "epochs": epochs,
    "alpha": alpha,
    }
)

wandb log

log를 남기고 wandb.ai/example/example에 접속하면 시작화된 log를 확인할 수 있다.

for epoch in range(epochs):
    loss, acc = train(train_loader)
    scheduler.step()
    
    wandb.log({"epoch": epoch, "learning rate": optimizer.param_groups[0]['lr'], "train loss": loss, "train accuracy": acc})