共用方式為


訓練 PyTorch 模型

PyTorch 是 Python 套件,可提供 GPU 加速的張量計算和高階功能,以組建深度學習網路。

MLflow PyTorch Notebook 適用於 MNIST 手寫數位辨識資料上的神經網路,並將結果記錄檔執行至 MLflow 伺服器。 它會在本機記錄 TensorFlow 事件格式的訓練計量和權數,然後將它們上傳至 MLflow 執行的成品目錄。 最後,它會啟動 TensorBoard,並讀取本機事件記錄。

當您準備好時,可使用搭配 Azure Databricks 的模型服務來部署您的模型。

MLflow PyTorch 模型訓練筆記本

取得筆記本