View a markdown version of this page

Cadres et algorithmes pris en charge - Amazon SageMaker AI

Les traductions sont fournies par des outils de traduction automatique. En cas de conflit entre le contenu d'une traduction et celui de la version originale en anglais, la version anglaise prévaudra.

Cadres et algorithmes pris en charge

Note

Amazon SageMaker Debugger n'est plus ouvert aux nouveaux clients. Les clients existants peuvent continuer à utiliser le service normalement. AWS continue d'investir dans l'amélioration de la sécurité et de la disponibilité de Debugger, mais nous ne prévoyons pas d'introduire de nouvelles fonctionnalités. Pour de plus amples informations, veuillez consulter Modification de la disponibilité du débogueur.

Le tableau suivant présente les frameworks et algorithmes d'apprentissage automatique de l' SageMaker IA pris en charge par Debugger.

SageMaker AI-supported frameworks and algorithms Debugging output tensors

TensorFlow

AWS TensorFlow conteneurs Deep Learning 1.15.4 ou version ultérieure

PyTorch

AWS PyTorch Deep Learning Containers 1.5.0 ou version ultérieure

MXNet

AWS Conteneurs de deep learning MXNet 1.6.0 ou version ultérieure

XGBoost

1.0-1, 1.2-1, 1.3-1

SageMaker IA générique ModelTrainer

Conteneurs de formation personnalisés (disponibles pour TensorFlow PyTorch, MXNet et XGBoost avec enregistrement manuel des crochets)

  • Débogage des tenseurs de sortie : suivez et déboguez les paramètres du modèle, tels que les pondérations, les gradients, les biais et les valeurs scalaires de votre tâche d'entraînement. Les frameworks d'apprentissage profond disponibles sont Apache MXNet, TensorFlow PyTorch, et XGBoost.

    Important

    Pour le TensorFlow framework avec Keras, SageMaker Debugger déconseille la prise en charge de zéro changement de code pour les modèles de débogage créés à l'aide des modules de la tf.keras version 2.6 et des versions ultérieures. TensorFlow Cela est dû aux modifications importantes annoncées dans la note de publication de la version TensorFlow 2.6.0. Pour obtenir des instructions de mise à jour de votre script d'entraînement, consultez Adaptez votre script TensorFlow d'entraînement.

    Important

    À partir de la PyTorch version 1.12.0, SageMaker Debugger déconseille la prise en charge du zéro changement de code pour les modèles de débogage.

    Cela est dû à des modifications importantes qui font que SageMaker Debugger interfère avec les torch.jit fonctionnalités. Pour obtenir des instructions de mise à jour de votre script d'entraînement, consultez Adaptez votre script PyTorch d'entraînement.

Si le framework ou l'algorithme que vous souhaitez entraîner et déboguer ne figure pas dans le tableau, rendez-vous sur le forum de AWS discussion et laissez des commentaires sur SageMaker Debugger.

Régions AWS

Amazon SageMaker Debugger est disponible dans toutes les régions où Amazon SageMaker AI est en service, à l'exception des régions suivantes.

  • Asie-Pacifique (Jakarta) : ap-southeast-3

Pour savoir si Amazon SageMaker AI est en service dans votre pays Région AWS, consultez la section Services AWS régionaux.

Utilisation de Debugger avec des conteneurs d’entraînement personnalisés

Transférez vos conteneurs de formation à l' SageMaker IA et obtenez des informations sur vos tâches de formation à l'aide de Debugger. Améliorez votre efficacité en optimisant votre modèle sur les instances Amazon EC2 à l'aide des fonctions de surveillance et de débogage.

Pour savoir comment créer votre conteneur d'entraînement avec la bibliothèque client sagemaker-debugger, le transmettre à Amazon Elastic Container Registry (Amazon ECR) et le contrôler et le déboguer, consultez Utilisation de Debugger avec des conteneurs d’entraînement personnalisés.

Référentiels de débogage Open-Source GitHub

Les API de débogage sont fournies via le SDK SageMaker Python et conçues pour créer des configurations de crochets et de règles de débogage pour les opérations d'IA et d'API. SageMaker CreateTrainingJob DescribeTrainingJob La bibliothèque client sagemaker-debugger fournit des outils pour enregistrer des hooks et accéder aux données d'entraînement via sa fonction d'essai, le tout grâce à ses opérations d'API flexibles et puissantes. Il prend en charge les frameworks d'apprentissage automatique TensorFlow PyTorch, MXNet et XGBoost sur Python 3.6 et versions ultérieures.

Pour obtenir des ressources directes sur Debugger et les opérations d'API sagemaker-debugger, consultez les liens suivants :

Si vous utilisez le SDK pour Java pour effectuer des tâches de SageMaker formation et que vous souhaitez configurer les API du débogueur, consultez les références suivantes :