调整训练脚本以使用 SMDDP 集体操作

本节提供的训练脚本示例经过简化，仅强调了在训练脚本中启用 SageMaker AI 分布式数据并行（SMDDP）库所需的更改。有关如何使用 SMDDP 库运行分布式训练作业的端到端 Jupyter Notebook 示例，请参阅 Amazon SageMaker AI 数据并行库示例。