Étape 1 : modifier votre propre script d'entraînement à l'aide de la bibliothèque de parallélisme de modèles distribués de SageMaker - Amazon SageMaker

Étape 1 : modifier votre propre script d'entraînement à l'aide de la bibliothèque de parallélisme de modèles distribués de SageMaker

Utilisez cette section pour savoir comment personnaliser votre script d'entraînement afin d'utiliser les fonctionnalités principales de la bibliothèque de parallélisme de modèles distribués Amazon SageMaker. Pour utiliser les fonctions et paramètres de l'API spécifique à la bibliothèque, nous vous recommandons d'utiliser cette documentation conjointement avec les API de la bibliothèque de parallélisme de modèles SageMaker dans la documentation relative au kit SDK Python SageMaker.

Les exemples de script d'entraînement fournis dans ces sections sont simplifiés et conçus pour mettre en évidence les modifications nécessaires à l'utilisation de la bibliothèque. Pour obtenir des exemples de blocs-notes exécutables de bout en bout illustrant l'utilisation d'un script d'entraînement TensorFlow ou PyTorch avec la bibliothèque de modèles parallèles distribués SageMaker, consultez Exemples de bloc-notes d'entraînement distribué Amazon SageMaker.