Condividi tramite


Eseguire il training di un modello PyTorch

PyTorch è un pacchetto Python che fornisce il calcolo del tensore con accelerazione GPU e funzionalità di alto livello per la creazione di reti di Deep Learning.

Il notebook MLflow PyTorch si adatta a una rete neurale nei dati di riconoscimento delle cifre scritti a mano MNIST e i log eseguono i risultati in un server MLflow. Registra le metriche e i pesi di training in formato di evento TensorFlow in locale e quindi li carica nella directory degli artefatti dell'esecuzione MLflow. Infine, avvia TensorBoard e legge gli eventi registrati a livello locale.

Quando si è pronti, è possibile distribuire il modello usando la gestione di modelli con Azure Databricks.

Notebook di training del modello PyTorch MLflow

Ottenere il notebook