Etapa 1: Adapte seu script de treinamento para registrar um hook - Amazon SageMaker

As traduções são geradas por tradução automática. Em caso de conflito entre o conteúdo da tradução e da versão original em inglês, a versão em inglês prevalecerá.

Etapa 1: Adapte seu script de treinamento para registrar um hook

O Amazon SageMaker Debugger vem com uma biblioteca cliente chamada Python sagemaker-debugger SDK. O sagemaker-debugger Python SDK fornece ferramentas para adaptar seu script de treinamento antes do treinamento e ferramentas de análise após o treinamento. Nesta página, você aprenderá como adaptar seu script de treinamento usando a biblioteca de cliente.

O sagemaker-debugger Python SDK fornece funções de wrapper que ajudam a registrar um hook para extrair os tensores do modelo, sem alterar seu script de treinamento. Para começar a usar a coleção de tensores de saída do modelo e depurá-los para encontrar problemas de treinamento, faça as seguintes modificações em seu script de treinamento.

dica

Enquanto estiver acompanhando esta página, use a documentação do SDK de código aberto do sagemaker-debugger para referências de API.