Configuración del depurador de SageMaker para guardar tensores

Los tensores son recopilaciones de datos de parámetros actualizados de los pasos hacia atrás y hacia adelante de cada iteración de entrenamiento. El depurador de SageMaker recopila los tensores de salida para analizar el estado de un trabajo de entrenamiento. Las operaciones del depurador de SageMaker CollectionConfig y las operaciones de la API DebuggerHookConfig proporcionan métodos para agrupar los tensores en colecciones y guardarlos en un bucket de 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.

Active el depurador de SageMaker AI especificando el parámetro debugger_hook_config mientras crea un estimador de SageMaker. 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.