wandb 로 log 남기기

서버 암호화 & 도커 사용이 원인으로 (추측되지만 사실 진짜 원인은 정확히는 모르겠다) 인해 텐서보드(tensorboard)를 사용하지 못하여 찾아본 것이 wandb (랩실에서 완드비라고 다들 발음하게되어서 그냥 완드비라고 말하는 것이 편하게 되었다) (Weights & Biases) 이다.

https://wandb.ai/site

Weights & Biases – Developer tools for ML

각종 log 값을 전부 잘 저장할 수가 있게 되어 매우 유용한데,

일단 가장 필요한 것은

 

wandb.init()

wandb.config.update({ ... }) # 이건 안해도 되긴 하더라

wandb.log({"epoch" : epoch값 저장하는 변수})
# 이건 필수, 없으면 몇 번째 epoch인지 몰라서 너무 불편함

wandb.log({"내가 저장하고 싶은 기록들 이름" : 내가 저장하고 싶은 기록들 값})

wandb.finish()

정도가 있는 것 같다.