MMF
WandbLogger 클래스는 Meta AI의 MMF 라이브러리에서 Weights & Biases에 트레이닝/검증 메트릭, 시스템(GPU 및 CPU) 메트릭, 모델 체크포인트 및 설정 파라미터를 로그하는 기능을 활성화합니다.
현재 기능
현재 MMF의 WandbLogger에 의해 지원되는 기능은 다음과 같습니다:
- 트레이닝 & 검증 메트릭
- 시간에 따른 학습률
- 모델 체크포인트를 W&B Artifacts에 저장
- GPU 및 CPU 시스템 메트릭
- 트레이닝 설정 파라미터
설정 파라미터
MMF 설정에서 wandb 로깅을 활성화하고 사용자 정의할 수 있는 다음 옵션들이 있습니다:
training:
    wandb:
        enabled: true
        
        # 엔티티는 runs을 보내는 사용자 이름 또는 팀 이름입니다.
        # 기본적으로 사용자 계정에 run을 로그합니다.
        entity: null
        
        # wandb와 실험을 로깅할 때 사용될 프로젝트 이름
        project: mmf
        
        # wandb와 프로젝트 아래에서 실험을 로깅할 때 사용될
        # 실험/ run 이름. 기본 실험 이름은: ${training.experiment_name}
        name: ${training.experiment_name}
        
        # 모델 체크포인팅을 켜고 W&B Artifacts에 체크포인트를 저장
        log_model_checkpoint: true
        
        # wandb.init()에 전달하고자 하는 추가 인수 값들.
        # 사용 가능한 인수가 무엇인지 보려면 https://docs.wandb.ai/ref/python/init
        # 문서를 확인하세요. 예를 들면:
        # job_type: 'train'
        # tags: ['tag1', 'tag2']
        
env:
    # wandb 메타데이터가 저장될 디렉토리의 경로를 변경하려면
    # (기본값: env.log_dir):
    wandb_logdir: ${env:MMF_WANDB_LOGDIR,}