wandb
-
[ML] Catboost wandb에 로그 올리기인공지능 2022. 5. 22. 03:09
Wandb(Weights & Biases) 란? Wandb 는 인공지능 실험(ML, DL)을 웹에서 학습과정을 모니터링 하고, 비교분석할 수 있는 Experiment Toolkit이다. 함께 언급되는 것은 Tensorboard등이 존재한다. Wandb는 Pytorch에서 사용시, 매 epoch을 반복할때마다 wandb.log를 사용하여 서버에 결과물을 업로드를 하는 형태를 취한다. import wandb wandb.init(config=args) model = ... # set up your model # Magic wandb.watch(model, log_freq=100) model.train() for batch_idx, (data, target) in enumerate(train_loader): o..