调整训练脚本以注册挂钩

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

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