[ML][PyTorch] WandB사용법

pip install wandb

wandb login

key를 wandb.ai에서 복사해서 붙여준다.

import wandb

 

코드의 첫 부분에 써준다.

wandb.init()

혹시 프로젝트 이름과 id를 명시하고싶다면

wandb.init(project='Machine-Learning-Project', entity='jonyejin')

이렇게 써주면 된다!

 

argparse가 선언이 되어있고, parse_args()로 args를 정의했다면 다음과 같이 wandb.config.update를 해준다.

def main():
    # Training settings
    parser = argparse.ArgumentParser(description='PyTorch MNIST Example')
    ...
    args = parser.parse_args()

wandb.config.update(args)

 

model 선언 후 다음과 같이 써준다.

wandb.watch(model)

 

train, val을 한번 돌린 후 다음과 같이 로그를 찍어준다.

wandb.log({
        "Examples": example_images,
        "Test Accuracy": 100. * correct / len(test_loader.dataset),
        "Test Loss": test_loss})

 

 

[출처]

https://greeksharifa.github.io/references/2020/06/10/wandb-usage/