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 sei pronto, puoi distribuire il tuo modello utilizzando Distribuire modelli usando il servizio Mosaic AI Model Serving.