모범 사례 - Amazon SageMaker AI

기계 번역으로 제공되는 번역입니다. 제공된 번역과 원본 영어의 내용이 상충하는 경우에는 영어 버전이 우선합니다.

모범 사례

다음 섹션에서는 파이프라인 단계에 @step 데코레이터를 사용할 때 따라야 할 모범 사례를 제안합니다.

웜 풀 사용

더 빠른 파이프라인 단계 실행을 위해 훈련 작업에 제공되는 웜 풀 기능을 사용합니다. 다음 코드 조각에 설명된 대로 @step 데코레이터에 keep_alive_period_in_seconds 인수를 제공하여 웜 풀 기능을 켤 수 있습니다.

@step( keep_alive_period_in_seconds=900 )

웜 풀에 대한 자세한 정보는 SageMaker AI Managed Warm Pools 부문을 참조하세요.

디렉터리 구조 지정

@step 데코레이터를 사용하는 동안 코드 모듈을 사용하는 것이 좋습니다. 단계 함수를 간접 호출하고 파이프라인을 정의하는 pipeline.py 모듈을 워크스페이스의 루트에 놓습니다. 권장하는 구조는 다음과 같습니다.

. ├── config.yaml # the configuration file that define the infra settings ├── requirements.txt # dependencies ├── pipeline.py # invoke @step-decorated functions and define the pipeline here ├── steps/ | ├── processing.py | ├── train.py ├── data/ ├── test/