Configurar SageMaker Debugger para guardar los tensores

Los tensores son recopilaciones de datos de parámetros actualizados de los pases hacia atrás y hacia adelante de cada iteración de entrenamiento. SageMaker El depurador recopila los tensores de salida para analizar el estado de un trabajo de entrenamiento. SageMaker Las operaciones del depurador CollectionConfigy de la DebuggerHookConfigAPI proporcionan métodos para agrupar los tensores en colecciones y guardarlos en un depósito S3 de destino. Los siguientes temas muestran cómo usar las operaciones de CollectionConfig y de la API de DebuggerHookConfig, seguido de ejemplos sobre cómo usar el enlace del depurador para guardar, acceder y visualizar los tensores de salida.

Al crear un estimador de SageMaker IA, active SageMaker Debugger especificando el parámetro. debugger_hook_config Los siguientes temas incluyen ejemplos de cómo configurar debugger_hook_config mediante las operaciones de la API de CollectionConfig y DebuggerHookConfig para extraer tensores de los trabajos de entrenamiento y guardarlos.