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