Träna en PyTorch-modell

PyTorch är ett Python-paket som tillhandahåller GPU-accelererad tensorberäkning och funktioner på hög nivå för att skapa djupinlärningsnätverk.

MLflow PyTorch-anteckningsboken passar ett neuralt nätverk på MNIST-handskrivna teckenigenkänningsdata och loggar kör resultat till en MLflow-server. Den loggar träningsmått och vikter i TensorFlow-händelseformat lokalt och laddar sedan upp dem till MLflow-körningens artefaktkatalog. Slutligen startar den TensorBoard och läser händelserna som loggas lokalt.

När du är redo kan du distribuera din modell med hjälp av modellservering med Azure Databricks.

Träningsanteckningsbok för MLflow PyTorch-modell

Hämta notebook-fil