Wytrenuj model PyTorch

PyTorch to pakiet języka Python, który zapewnia przyspieszone przez procesor GPU obliczenia tensorowe i funkcje wysokiego poziomu do tworzenia sieci uczenia głębokiego.

Notes MLflow PyTorch pasuje do sieci neuronowej na danych rozpoznawania cyfr odręcznych MNIST i dzienniki uruchamiają wyniki na serwerze MLflow. Rejestruje metryki trenowania i wagi w formacie zdarzenia TensorFlow lokalnie, a następnie przekazuje je do katalogu artefaktów przebiegu MLflow. Na koniec uruchamia tensorBoard i odczytuje zdarzenia zarejestrowane lokalnie.

Gdy wszystko będzie gotowe, możesz wdrożyć model przy użyciu usługi Model obsługujących usługę Azure Databricks.

Notes trenowania modelu MLflow PyTorch

Pobierz notes