本文属于机器翻译版本。若本译文内容与英语原文存在差异,则一律以英文原文为准。
最佳实操
以下各节建议了在工作流步骤中使用@step
装饰器时应遵循的最佳实践。
使用温水池
要更快地运行管道步骤,请使用为训练作业提供的热池功能。你可以通过向@step
装饰器提供keep_alive_period_in_seconds
参数来开启暖池功能,如以下片段所示:
@step( keep_alive_period_in_seconds=900 )
有关暖池的更多信息,请参阅 使用 SageMaker 托管的暖池进行训练。
整理你的目录
建议您在使用@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/