Configuration de SageMaker Debugger pour enregistrer les tenseurs

Les tenseurs sont des collections de données de paramètres mis à jour à partir de la passe en arrière et en avant de chaque itération d'entraînement. SageMaker Debugger collecte les tenseurs de sortie pour analyser l'état d'une tâche d'entraînement. Les opérations d'API CollectionConfig et DebuggerHookConfig de SageMaker Debugger fournissent des méthodes pour regrouper les tenseurs dans des collections et les enregistrer dans un compartiment S3 cible. Les rubriques suivantes montrent comment utiliser les opérations CollectionConfig et DebuggerHookConfig API, et sont suivies d’exemples sur l’utilisation du hook de Debugger pour accéder aux tenseurs de sortie, les visualiser et les enregistrer.

Lorsque vous construisez un estimateur SageMaker AI, activez SageMaker Debugger en spécifiant le paramètre debugger_hook_config. Les rubriques suivantes incluent des exemples de configuration de debugger_hook_config à l’aide des opérations d’API CollectionConfig et DebuggerHookConfig pour extraire les tenseurs de vos tâches d’entraînement et les enregistrer.