Tensorboard / Weight&Bias

yoonene·2022년 3월 3일
0

Boostcamp AI Tech

목록 보기
25/27

Tensorboard

pip install tensorboard
tensorflow --logdir [저장된 모델 위치]--port [부여받은 포드번호] --host 0.0.0.0

Weight & Bias

import wandb
wandb.init(project="project_name")
config = wandb.config
config.learning_rate = args.learning_rate
wandb.watch(model)
wandb.log({"loss": loss})
wandb.log({"lr": args.learning_rate})
profile
NLP Researcher / Information Retrieval / Search

1개의 댓글

comment-user-thumbnail
2022년 3월 7일

mlflow도 한 번 사용하고 비교해보면 좋을 것 같습니다!

답글 달기