WandbLogger 클래스는 학습/검증 지표, 시스템(GPU 및 CPU) 지표, 모델 체크포인트 및 구성 파라미터를 W&B로 로깅할 수 있도록 해줍니다.
현재 기능
WandbLogger에서 지원됩니다.
- 학습 및 검증 지표
- 시간에 따른 학습률
- 모델 체크포인트를 W&B 아티팩트에 저장
- GPU 및 CPU 시스템 메트릭
- 학습 설정 파라미터
W&B를 Meta AI의 MMF와 통합하는 방법.
WandbLogger 클래스는 학습/검증 지표, 시스템(GPU 및 CPU) 지표, 모델 체크포인트 및 구성 파라미터를 W&B로 로깅할 수 있도록 해줍니다.
WandbLogger에서 지원됩니다.
training:
wandb:
enabled: true
# 엔터티는 실행을 전송할 사용자 이름 또는 팀 이름입니다.
# 기본적으로 사용자 계정에 실행을 기록합니다.
entity: null
# wandb로 실험을 기록할 때 사용할 프로젝트 이름
project: mmf
# wandb로 프로젝트 하위에 실험을 기록할 때 사용할
# 실험/실행 이름. 기본 실험 이름:
# ${training.experiment_name}
name: ${training.experiment_name}
# 모델 체크포인팅을 활성화하여 체크포인트를 W&B 아티팩트에 저장
log_model_checkpoint: true
# wandb.init()에 전달할 추가 인수 값 (예:
# job_type: 'train'
# tags: ['tag1', 'tag2']
env:
# wandb 메타데이터가 저장될 디렉토리 경로를 변경하려면
# (기본값: env.log_dir):
wandb_logdir: ${env:MMF_WANDB_LOGDIR,}