步骤 1:调整训练脚本以注册钩子 - Amazon SageMaker
Amazon Web Services 文档中描述的 Amazon Web Services 服务或功能可能因区域而异。要查看适用于中国区域的差异,请参阅 中国的 Amazon Web Services 服务入门 (PDF)

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

步骤 1:调整训练脚本以注册钩子

Amazon D SageMaker ebugger 附带了一个名为 sagemaker-debuggerPython SDK 的客户端库。sagemaker-debugger Python SDK 提供了用于在训练之前调整训练脚本的工具,以及在训练后使用的分析工具。在本页面中,您将学习如何使用客户端库调整训练脚本。

sagemaker-debugger Python SDK 提供了包装器,可协助注册钩子来提取模型张量,而无需更改训练脚本。要开始收集模型输出张量并对其进行调试以发现训练问题,请在训练脚本中进行以下修改。

提示

在按照本页面的说明操作时,请使用 sagemaker-debugger 开源 SDK 文档获取 API 参考。