W&B 사용법

[ilovemyminutes]

계정 등록

I. 사이트 로그인

• Weight and Bias에서 로그인을 먼저 해주세요!
• 공유 구글 계정으로 로그인합니다.
• 공유 아이디: [email protected]
• 비밀번호: smblysmbly!

II. 로컬 작업 환경에 계정 등록

• wandb가 설치되지 않았다면 설치부터 합시다!

  $ pip install wandb
  

• 터미널에 다음을 입력하여 wandb를 활성화합니다

  $ wandb init
  

• 그러면 다음과 같은 선택지가 나올텐데, (2)를 선택해주세요

  (1) Create a W&B account
  (2) Use an existing W&B account # 요녀석을 선택
  (3) Don't visualize my results
  

• API 키를 입력하라는 메시지가 나오면서 authorize 머시기머시기 링크를 주는데, 해당 링크로 접속한 뒤 API 키를 복사합니다.
• 입력란에 복사한 키를 입력해주면 등록 끝!

모델 학습 시 추가된 Arguments

• train.py에 다음과 같이 --project_name, --exp_name argument가 추가되었습니다. 모두 W&B 로깅을 위해 필요한 arguement입니다!
• --project_name: W&B에 등록할 프로젝트 이름을 설정합니다. (MathOCR-SATRN-튜닝 등)
• --exp_name: 등록한 W&B 프로젝트 내에 등록될 실험명을 설정합니다. (SATRN-베이스라인, SATRN-AdamP사용 등)

parser.add_argument(
        '--project_name',
        default='MathOCR-iloveslowfood',
        help='WandB에서 사용할 자신의 프로젝트 이름(MathOCR-준구의실험교실 등)'
    ) # 추가된 것
parser.add_argument(
    '--exp_name',
    default='semi-aster',
    help='실험 이름(SATRN-베이스라인, SARTN-Loss변경 등)'
) # 추가된 것
parser.add_argument(
    "-c",
    "--config_file",
    dest="config_file",
    default="./configs/Attention.yaml",
    type=str,
    help="Path of configuration file",
) # 원래 있던 것