Apprentissage d’un modèle PyTorch
PyTorch est un package Python qui fournit une fonctionnalité de calcul de tenseur accélérée par GPU et des fonctionnalités générales pour la création de réseaux Deep Learning.
Le notebook MLflow PyTorch applique un réseau neuronal à des données de reconnaissance de chiffres manuscrits MNIST et journalise les résultats des exécutions sur un serveur MLflow. Il journalise les pondérations et les métriques d’entraînement dans le format d’événement TensorFlow localement, puis les charge sur le répertoire d’artefacts de l’exécution de MLflow. Enfin, il démarre TensorBoard et lit les événements journalisés localement.
Quand vous êtes prêt, vous pouvez déployer votre modèle en utilisant Service de modèle avec Azure Databricks.