火車PyTorch模型

PyTorch是一個Python包,提供GPU-accelerated張量計算和高水平的功能構建深度學習網絡。

的MLflow PyTorch筆記本適合神經網絡在MNIST手寫數字識別數據和日誌MLflow服務器運行結果。這日誌訓練在本地TensorFlow事件格式的指標和權重,然後上傳MLflow運行的構件目錄。最後,它開始TensorBoard並在本地讀取事件記錄。