Melatih model PyTorch
PyTorch adalah paket Python yang menyediakan komputasi tensor yang dipercepat-GPU dan fungsionalitas tingkat tinggi untuk membangun jaringan pembelajaran mendalam.
Notebook MLflow PyTorch cocok dengan jaringan saraf pada data pengenalan digit tulisan tangan MNIST dan mencatat hasil run ke server MLflow. Notebook ini mencatat metrik dan bobot pelatihan dalam format acara TensorFlow secara lokal dan kemudian mengunggahnya ke direktori artefak MLflow run. Terakhir notebook ini memulai TensorBoard dan membaca peristiwa yang dicatat secara lokal.
Setelah siap, Anda dapat menyebarkan model menggunakan Penyajian model dengan Azure Databricks.