Skip to main content
W&B는 Amazon SageMaker와 통합되어 하이퍼파라미터를 자동으로 읽고, 분산 runs를 그룹화하고, 체크포인트에서 runs를 재개합니다.

​
인증

W&B는 트레이닝 스크립트를 기준으로 한 상대 경로에 있는 secrets.env 파일을 찾고, wandb.init()이 호출될 때 그 내용을 환경 변수로 로드합니다. 실험을 실행할 때 사용하는 스크립트에서 wandb.sagemaker_auth(path="source_dir")를 호출하면 secrets.env 파일을 생성할 수 있습니다. 이 파일을 반드시 .gitignore에 추가하세요!

​
기존 Estimator

SageMaker에서 미리 구성해 둔 Estimator 중 하나를 사용하는 경우, 소스 디렉터리에 wandb를 포함한 requirements.txt 파일을 추가해야 합니다. 
wandb
Python 2에서 실행되는 estimator를 사용 중이라면, wandb를 설치하기 전에 이 wheel을 통해 psutil을 직접 설치해야 합니다: 
https://wheels.galaxyproject.org/packages/psutil-5.4.8-cp27-cp27mu-manylinux1_x86_64.whl
wandb
GitHub에서 전체 예제를 확인하고, 블로그에서 자세한 내용을 읽어보세요. 또한 SageMaker와 W&B를 사용해 감성 분석기를 배포하는 방법을 설명한 Deploy Sentiment Analyzer Using SageMaker and W&B 튜토리얼도 참고할 수 있습니다.
W&B sweep 에이전트는 SageMaker 인테그레이션이 꺼져 있을 때에만 SageMaker 작업에서 예상대로 동작합니다. wandb.init 호출을 수정하여 SageMaker 인테그레이션을 끄세요:
wandb.init(..., settings=wandb.Settings(sagemaker_disable=True))