訓練 PyTorch 模型
PyTorch 是 Python 套件,可提供 GPU 加速的張量計算和高階功能,以組建深度學習網路。
MLflow PyTorch Notebook 適用於 MNIST 手寫數位辨識資料上的神經網路,並將結果記錄檔執行至 MLflow 伺服器。 它會在本機記錄 TensorFlow 事件格式的訓練計量和權數,然後將它們上傳至 MLflow 執行的成品目錄。 最後,它會啟動 TensorBoard,並讀取本機事件記錄。
當您準備好時,可使用搭配 Azure Databricks 的模型服務來部署您的模型。