本文為英文版的機器翻譯版本,如內容有任何歧義或不一致之處,概以英文版為準。
最佳實務
以下各節建議在您針對管道步驟使用 @step 裝飾項目時要遵循的最佳實務。
使用暖集區
如需加快管道步驟的執行速度,請使用為訓練任務提供的暖集區功能。您可以將 keep_alive_period_in_seconds 引數提供給 @step 裝飾項目以開啟暖集區功能,如下列程式碼片段所示範:
@step( keep_alive_period_in_seconds=900 )
如需有關暖集區的詳細資訊,請參閱 SageMaker AI 受管暖集區。
建構您的目錄
建議您在使用 @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/