Tutoriel : Modèles Deep Learning de bout en bout sur Azure Databricks

Ce notebook de tutoriel présente un exemple de bout en bout d’apprentissage d’un modèle d’apprentissage profond dans Azure Databricks, notamment le chargement de données, la visualisation des données, la configuration d’une optimisation d’hyperparamètre parallèle et l’utilisation de MLflow pour examiner les résultats, inscrire le modèle et effectuer une inférence sur de nouvelles données à l’aide du modèle inscrit dans une UDF Spark.

Le notebook utilise PyTorch, un package Python qui fournit un calcul tensoriel accéléré par GPU et des fonctionnalités de haut niveau pour la création de réseaux d’apprentissage profond.

Lorsque vous êtes prêt, vous pouvez déployer votre modèle en utilisant Mosaic AI Model Serving.

Notebook d’entraînement de modèle MLflow PyTorch

Obtenir le notebook