MMF

Meta AI의 MMF 프레임워크와 W&B를 통합하여 멀티모달 모델 트레이닝 실험을 추적하고 메트릭을 기록합니다.

Meta AI의 MMF 라이브러리의 WandbLogger 클래스는 W&B가 트레이닝/검증 메트릭, 시스템(GPU 및 CPU) 메트릭, 모델 체크포인트, 그리고 설정 매개변수를 기록할 수 있게 해줍니다.

현재 지원되는 특성

MMF의 WandbLogger는 현재 다음 특성을 지원합니다:

• 트레이닝 및 검증 메트릭
• 시간에 따른 학습률
• 모델 체크포인트를 W&B Artifacts에 저장
• GPU 및 CPU 시스템 메트릭
• 트레이닝 설정 매개변수

설정 매개변수

다음 옵션을 MMF 설정에서 wandb 로깅을 활성화하고 사용자 지정하는 데 사용할 수 있습니다:

training:
    wandb:
        enabled: true
        
        # entity는 run을 전송할 사용자 이름 또는 팀 이름입니다.
        # 기본적으로 사용자 계정에 run을 기록합니다.
        entity: null
        
        # wandb로 실험을 기록할 때 사용할 프로젝트 이름
        project: mmf
        
        # wandb로 프로젝트 하위에 실험을 기록할 때 사용할
        # 실험/run 이름. 기본 실험 이름:
        # ${training.experiment_name}
        name: ${training.experiment_name}
        
        # 모델 체크포인팅을 활성화하여 체크포인트를 W&B Artifacts에 저장
        log_model_checkpoint: true
        
        # wandb.init()에 전달할 추가 argument 값 예시:
        # job_type: 'train'
        # tags: ['tag1', 'tag2']
        
env:
    # wandb 메타데이터가 저장될 디렉터리 경로를 변경하려면
    # (기본값: env.log_dir):
    wandb_logdir: ${env:MMF_WANDB_LOGDIR,}