[Wandb] 손쉽게 실험 결과 기록 및 실험 환경 보존

딥러닝 실험을 진행하다 보면 학습이 잘 되는지 안 되는지 중간중간 모니터링 할 필요가 있다. 그래서 텐서 대시보드 등 다양한 툴이 존재하는데 이번 STS 대회를 진행하면서 Wandb(Weight & Biases)를 사용해봤는데 매우 유용한 툴이다. 개인용으로는 무료로 사용할 수 있어 접근성 또한 편리하며 협업하기 매우 쉽다. 우선 여기에서 회원가입을 진행하자. 또한, pytorch lightning 라이브러리랑 연결해서 사용하는 방법을 공유하겠다.

1. Wandb 설치

pip install wandb

2. 코드 작성

import wandb
import pytorch_lightning as pl

# wandb 설정
wandb.init(name='{실험명}', project='{프로젝트 이름}', entity='{계정 이름}')
wandb_logger = WandbLogger(save_dir='{저장 경로}')
wandb_logger.experiment.config.update({config 변수})

pl.Trainer(..., logger = wandb_logger) # log 연결