使用 SageMaker 的分布式模型并行库修改训练脚本 - Amazon SageMaker
Amazon Web Services 文档中描述的 Amazon Web Services 服务或功能可能因区域而异。要查看适用于中国区域的差异,请参阅中国的 Amazon Web Services 服务入门

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

使用 SageMaker 的分布式模型并行库修改训练脚本

使用本部分了解如何使用 Amazon SageMaker 的分布式模型并行库特定 API 函数和参数自定义您的培训脚本。建议您将本文档与模型并行库 API 文档.

这些部分中提供的培训脚本示例经过简化,旨在突出显示使用库时必须进行的必要更改。有关演示如何在 SageMaker 分布式模型并行库中使用 TensorFlow 或 PyTorch 训练脚本的端到端、可运行的笔记本示例,请参阅Amazon SageMaker 分布式培训笔记本示例.