1단계: 훈련 스크립트를 조정하여 후크 등록

Amazon SageMaker 디버거에는 sagemaker-debuggerPython SDK라는 클라이언트 라이브러리가 함께 제공됩니다. sagemaker-debugger Python SDK는 훈련 전에 훈련 스크립트를 조정하는 도구와 훈련 후에 분석 도구를 제공합니다. 이 페이지에서는 클라이언트 라이브러리를 사용하여 훈련 스크립트를 조정하는 방법을 알아봅니다.

sagemaker-debugger Python SDK는 훈련 스크립트를 변경하지 않고도 모델 텐서를 추출하기 위한 후크를 등록하는 데 도움이 되는 래퍼 함수를 제공합니다. 모델 출력 텐서 컬렉션을 시작하고 이를 디버깅하여 훈련 문제를 찾으려면 훈련 스크립트를 다음과 같이 수정하세요.