wandb 사용법 설치 로그인 시작

 

wandb는 머신러닝 실험을 관리하고 추적하는 도구이다. 실험의 하이퍼파라미터, 메트릭, 로그, 시각화 등을 웹 대시보드를 통해 모니터링할 수 있다. 개발자가 여러 실험을 쉽게 비교하고 재현할 수 있도록 도와준다.

1. 설치 및 로그인

pip install wandb

wandb login

설치하고 로그인 한다.

 

API key 복사해서 입력(https://app.wandb.ai/authorize)

2. relogin

relogin은 다음 명령어 치고 똑같이 key 입력

~$ wandb login --relogin
wandb: Logging into wandb.ai. (Learn how to deploy a W&B server locally: https://wandb.me/wandb-server)
wandb: You can find your API key in your browser here: https://wandb.ai/authorize
wandb: Paste an API key from your profile and hit enter, or press ctrl+c to quit: 
wandb: Appending key for api.wandb.ai to your netrc file: /mnt/data/minsun3054/.netrc
wandb: W&B API key is configured. Use `wandb login --relogin` to force relogin

 

3. 사용법

대충 기억에 남는 사용법 여러가지 코드로 기록

wandb.log()로그찍는거는 dictionary 형식이어야함

wandb.init(project = "ㅇㅇㅇ", name ="1 image")

        wandb.log({"MPI Full Image": wandb.Image(mpi_full_image, caption=f"Epoch {epoch}")})

        wandb.log({
            "Epoch": epoch + 1,
            "Epoch Progress": f"{epoch+1}/{num_epochs}",
            "Loss": epoch_loss / len(dataloader)
        })
# ✅ W&B 종료
wandb.finish()


#참고
wandb.init(project="생성한 project 이름", config={
    "learning_rate": 0.0001,
    "dropout": 0.2,
    "architecture": "CNN",
    "dataset": "COCO",
})
config = wandb.config
wandb.log({"acc":acc, "loss":loss})

 

 

 

Reference

https://mingyu6952.tistory.com/entry/wandb-%EC%82%AC%EC%9A%A9%EB%B2%95Pytorch-CIFAR10-%EB%B6%84%EB%A5%98

https://iambeginnerdeveloper.tistory.com/206