Adapting your training script to register a hook

Amazon SageMaker Debugger comes with a client library called the sagemaker-debugger Python SDK. The sagemaker-debugger Python SDK provides tools for adapting your training script before training and analysis tools after training. In this page, you'll learn how to adapt your training script using the client library.

The sagemaker-debugger Python SDK provides wrapper functions that help register a hook to extract model tensors, without altering your training script. To get started with collecting model output tensors and debug them to find training issues, make the following modifications in your training script.