[꿀팁] Wandb 사용하기

[JAEWOOSUN]

1. Wandb 홈페이지 회원가입

https://wandb.ai/site

2. Wandb library 설치

  pip install wandb

3. linux에 wandb 로그인

  wandb login
  
  # 결과:
  wandb: You can find your API key in your browser here: https://app.wandb.ai/authorize
  wandb: Paste an API key from your profile and hit enter:

이후 https://app.wandb.ai/authorize <- 여기 링크에 들어가서 API key를 복사하면 됩니다!

4. Pytorch code 수정

• wandb를 import합니다.

  import wandb
  

• main() 함수의 맨 앞부분에 다음 코드를 추가합니다.

  wandb.init()
  

• args 변수 선언부 밑에 다음 코드를 추가합니다.

  wandb.config.update(args)
  

• model 선언부 다음에 다음 코드를 추가합니다.

  wandb.watch(model)
  

• log를 추적할 수 있도록 eval_result 값을 추가합니다.

  wandb.log({
            "joint_goal_accuracy": eval_result['joint_goal_accuracy'],
            "turn_slot_accuracy": eval_result['turn_acc_score'],
            "turn_slot_f1": eval_result['slot_F1_score']
        })
  

5. Reference

추가적인 다양한 기능들을 확인할 수 있습니다!
https://greeksharifa.github.io/references/2020/06/10/wandb-usage/