[wandb] wandb 사용해서 그래프로 보기

tensorboard 보다 더 자세하고, 어디에서든 인터넷 접속만 하면 볼 수 있는 실시간 그래프를 그려주는..!

wandb 사용법을 알아보자 ㅋㅋㅋ

pip install wandb

로그인을 해주어야 하는데, 아래 이미지와 같이 명령어(“wandb login“)를 입력해주고 링크를 따라간다.

주소를 따라가면 아래와 같이 API Key가 나온다! 복사해준 후, 그대로! 입력을 해준다

그럼 따란! 준비완료!

근데 나는 위 그림 처럼 나오는데, 다른 블로그에서는 아래와 같은 문구가 나온단다.. 아마 나는 여러번 썼기 때문에 그런게 아닐까?!

Successfully logged in to Weights & Biases!

다시 페이지 들어가보면 이렇게 나의 프로필이 연동되어서 생기고(나는 github으로 가입하여서 github연동됨), 그동안 한 프로젝트들이 정리되어져 볼 수 있다.

4-1) wandb import 먼저!

import wandb

4-2) main() 함수의 맨 앞부분에 init() 으로 초기화해준다! 혹, main()을 따로 하지 않고 바로 파일에 박아버릴 경우엔 아래 예시와 같이 그냥 써주면 된다!

wandb.init()

4-3) args 변수 선언부 밑에 아래와 같이 추가해준다.

wandb.config.update(args)

4-4) model 선언 다음에 다음 코드를 추가한다.

wandb.watch(model)

예시!

4-5) 마지막으로 wandb.log 를 작성해준다. 나는 loss 계산하는 파일이 따로 생성되어있어서 다음과 같이 해줌!

파일을 실행하면 다음과 같이 친절하게 graph를 볼 수 있는 URL 주소를 준다 ㅎㅎ

view run at http://~ 이 주소를 클릭하거나 복사해서 인터넷으로 보면 됩니다~

음..! 어찌저찌 학습이 되나봅니다.

이제 언제 어디서든 학습 상태를 확인해 볼 수 있도록 해요~🙂🙂🙂

Svg_test