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 będziesz gotowy, możesz wdrożyć swój model przy użyciu Wdrażanie modeli przy użyciu usługi Mosaic AI Model Serving.