Configuração do SageMaker Debugger para salvar tensores

Os tensores são coleções de dados de parâmetros atualizados da passagem para trás e para frente de cada iteração de treinamento. O SageMaker Debugger coleta os tensores de saída para analisar o estado de um trabalho de treinamento. As operações do SageMaker Debugger CollectionConfig e DebuggerHookConfig da API fornecem métodos para agrupar tensores em coleções e salvá-los em um bucket do S3 de destino. Os tópicos a seguir mostram como usar as operações da API CollectionConfig e DebuggerHookConfig, seguidos por exemplos de como usar o hook do Depurador para salvar, acessar e visualizar tensores de saída.

Quando criar um estimador do SageMaker AI, ative o SageMaker Debugger especificando o parâmetro debugger_hook_config. As etapas a seguir incluem exemplos de como configurar debugger_hook_config usando as operações da API CollectionConfig e DebuggerHookConfig para retirar tensores de seus trabalhos de treinamento e salvá-los.