最佳实践 - Amazon SageMaker

本文属于机器翻译版本。若本译文内容与英语原文存在差异,则一律以英文原文为准。

最佳实践

以下各节建议了在工作流步骤中使用@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/