딥러닝 | Deep Learning

wandb(weight & biases) 사용법

sungkshon 2024. 8. 18. 04:41
반응형

👉 wandb(weight & biases)란?

딥러닝 실험 과정을 손쉽게 tracking하고, 더 나은 모델을 빨리 만들 수 있도록 도와주는, 시각화 할 수 있는 tool이다. 

wandb는 머신러닝 실험을 위한 대시보드 시각화, 파라미터 튜닝 자동화 등 다양한 기능을 제공합니다.

wandb platform은 5가지 유용한 도구를 제공하고 있다.

wandb platform

  • Experiments : 머신러닝 모델 실험을 추적하기 위한 dashbord 제공
  • Artifacts : Dataset version관리와 Model version관리
  • Tables : Data를 loging하여 wandb로 시각화하고 query하는데 사용
  • sweeps : Hyper-parameter를 자동으로 tuning하여 최적화함
  • Reports : 실험을 document로 정리하여 collaborators와 공유

무료버전은 개인 프로젝트만 관리가 가능하다. 100GB의 저장공간을 지원해준다. 

web UI는 웹 브라우저에서 모든 기능을 이용할 수 있다.

👉초기 설정

1. wandb 회원가입(https://wandb.ai/site)

2. wandb install

pip install wandb

 3. 로그인

wandb login

 설정에 들어가 API key를 복사한 다음 명령창에 붙여넣기를 해주면 로그인이 완료된다. 

 

반응형