Обучение модели PyTorch
PyTorch — это пакет Python, обеспечивающий тензорное вычисление с ускорением GPU и высокоуровневые функции для создания сетей глубокого обучения.
Записная книжка MLflow PyTorch создана для нейронной сети на основе результатов распознавания рукописных цифр MNIST и ведет журнал результатов запусков на сервере MLflow. Она регистрирует метрики обучения и весовые коэффициенты локально в формате событий TensorFlow, а затем передает их в каталог артефактов запуска MLflow. Наконец, она запускает TensorBoard и считывает зарегистрированные локально события.
Когда вы будете готовы, вы можете развернуть вашу модель с помощью сервиса развертывания моделей Mosaic AI.